PyPI page
Home page
Author:
Marco Simoni
License:
MIT
Summary:
One JAX/Flax backbone for AR, discrete-diffusion, and flow-matching LLMs — switch paradigm with a single config field.
Latest version:
0.4.14
Required dependencies:
flax
|
jax
|
jaxlib
|
msgpack
|
numpy
|
optax
|
pyyaml
|
tokenizers
|
tqdm
Optional dependencies:
build
|
dantinox
|
datasets
|
huggingface-hub
|
interrogate
|
matplotlib
|
mike
|
mkdocs-glightbox
|
mkdocs-material
|
mkdocs-minify-plugin
|
mkdocstrings
|
mypy
|
numpy
|
pandas
|
pre-commit
|
pytest
|
pytest-cov
|
ruff
|
scipy
|
transformers
|
twine
|
types-pyyaml
|
types-tqdm
Downloads last day:
7
Downloads last week:
29
Downloads last month:
106
7-day smoothing