PyPI page
Home page
Author:
None
License:
Apache-2.0
Summary:
A lightweight JAX-native LLM post-training framework.
Latest version:
0.1.6
Required dependencies:
datasets
|
flax
|
fsspec
|
google-metrax
|
grain
|
hf_transfer
|
huggingface_hub
|
jaxtyping
|
jinja2
|
kagglehub
|
numba
|
omegaconf
|
perfetto
|
pillow
|
pylatexenc
|
python-dotenv
|
qwix
|
sentencepiece
|
sympy
|
tenacity
|
tensorflow_datasets
|
tqdm
|
transformers
Optional dependencies:
ipython
|
jax
|
matplotlib
|
myst-nb
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-book-theme
|
sphinx-collections
|
sphinx-gallery
|
sphinx_contributors
Downloads last day:
1,699
Downloads last week:
7,387
Downloads last month:
64,126