transformers==4.41.1
torch==2.4.0
torch-xla[tpu]==2.4.0
typer==0.6.1
loguru==0.6.0
sentencepiece==0.2.0

[jetstream-pt]
jetstream-pt
torch-xla[pallas]==2.4.0

[quality]
black
ruff
isort

[tests]
pytest
safetensors
