transformers==4.46.3
torch==2.5.1
torch-xla[tpu]==2.5.1
typer==0.6.1
loguru==0.6.0
sentencepiece==0.2.0

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

[quality]
black
ruff
isort

[tests]
pytest
safetensors
