PyPI page
Home page
Author:
None
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-gallery
|
sphinx_contributors
|
tensorflow
Downloads last day:
4,401
Downloads last week:
31,858
Downloads last month:
157,856