pyarabic~=0.6
datasets~=3.6
python-dotenv~=1.1
jiwer~=3.1
dspy~=2.6
tabulate~=0.9
sentencepiece~=0.2
transformers[torch]~=4.52
pytorch-lightning>=2.6.0
torchmetrics>=1.8.2
scikit-learn>=1.6.1

[cpu]
torch>=2.6.0

[cuda]
torch>=2.6.0

[dev]
ruff
mypy<2.0,>=1.0
black<25.0,>=23.0
isort<5.14,>=5.12
pytest
pytest-cov
wandb~=0.20
matplotlib>=3.9.4
matplotlib-inline>=0.2.1
seaborn~=0.13
ipython>=8.18.1
kenlm
