PyPI page
Author:
Tunix Developers
License:
Apache-2.0
Summary:
A lightweight JAX-native LLM post-training framework.
Latest version:
0.1.7
Required dependencies:
datasets
|
flax
|
fsspec
|
google-metrax
|
grain
|
hf-transfer
|
huggingface-hub
|
importlib-resources
|
jaxtyping
|
jinja2
|
kagglehub
|
numba
|
omegaconf
|
orbax-checkpoint
|
perfetto
|
pillow
|
pylatexenc
|
python-dotenv
|
qwix
|
safetensors
|
sentencepiece
|
sympy
|
tenacity
|
tensorflow-datasets
|
tqdm
|
transformers
Optional dependencies:
chex
|
gcsfs
|
ipython
|
jax
|
matplotlib
|
myst-nb
|
pytest
|
pytest-xdist
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-book-theme
|
sphinx-collections
|
sphinx-contributors
|
sphinx-gallery
|
tensorflow
Downloads last day:
3,986
Downloads last week:
28,054
Downloads last month:
235,166
7-day smoothing