piwheels

google-tunix

A lightweight JAX-native LLM post-training framework.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install google-tunix

Dependencies

• grain
• tensorflow-datasets
• kagglehub
• tqdm
• tensorboardx
• gcsfs
• python-dotenv
• transformers
• omegaconf
• jaxtyping
• sentencepiece
• datasets
• qwix
• jinja2
• huggingface-hub

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.1.2	2025-10-10
google_tunix-0.1.2-py3-none-any.whl (234 KB)
0.1.1	2025-10-08
google_tunix-0.1.1-py3-none-any.whl (227 KB)
0.1.0	2025-09-30
google_tunix-0.1.0-py3-none-any.whl (218 KB)
0.1.0.dev1 pre-release	2025-09-30
google_tunix-0.1.0.dev1-py3-none-any.whl (218 KB)
0.1.0.dev0 pre-release	2025-09-30
google_tunix-0.1.0.dev0-py3-none-any.whl (218 KB)
0.0.2	2025-09-24
google_tunix-0.0.2-py3-none-any.whl (207 KB)
0.0.1	2025-09-20

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue