-r requirements.txt

# Training and checkpoint export dependencies
numpy
jax
jaxlib
flax>=0.10.2
optax
safetensors
sentencepiece
