datasets
gcsfs
grain
huggingface_hub
jax
jax[tpu]
jaxtyping
kagglehub
omegaconf
qwix
sentencepiece
tensorboardX
tensorflow_datasets
tqdm
transformers
python-dotenv

[dev]
flax>=0.11.2

[docs]
sphinx>=8.2.3
sphinx-book-theme>=1.1.4
sphinx-autodoc-typehints
ipython>=8.8.0
myst-nb>=1.3.0
matplotlib>=3.10.0
sphinx-gallery>=0.19.0
sphinx-collections>=0.0.1
sphinx_contributors

[prod]
flax>=0.12.0
