jax>=0.4.31
jaxlib>=0.4.31
optax>=0.2.2
matplotlib>=3.8.2
numpy>=1.24.0
einops>=0.8.0
tqdm>=4.66.1
jax-tqdm>=0.1.2
flax>=0.12.0
chex>=0.1.86
jaxtyping>=0.2.24
Pillow>=10.0.0
huggingface-hub>=0.20.0
myst-nb>=1.3.0

[dev]
pytest>=6.0
pytest-cov
pytest-xdist
nbmake>=1.5.0

[docs]
sphinx>=7.1.0
sphinx-book-theme>=1.0.0
sphinx-autodoc-typehints>=1.24.0
sphinx-copybutton>=0.5.2
sphinx-design>=0.5.0
sphinx-togglebutton>=0.3.0
myst-parser>=2.0.0
myst-nb>=1.0.0
linkify-it-py>=2.0.0
ipython>=8.0.0
sphinxcontrib-tikz>=0.4.0
sphinxcontrib-bibtex>=2.6.0
sphinxext-opengraph>=0.8.2

[flux]
sentencepiece>=0.1.99
tokenizers>=0.15.0
orbax-checkpoint>=0.5.0
