jax==0.11.2
equinox>=0.13
optax>=0.2.8
numpy>=2
safetensors>=0.4

[cuda]
jax[cuda13]==0.11.2

[eval]
lm-eval

[loggers]
wandb
tensorboardX

[orbax]
orbax-checkpoint

[stream]
datasets
pyarrow

[test]
pytest>=8

[tokenize]
transformers
jinja2
