torch>=1.9.0
numpy>=1.21.0
matplotlib>=3.4.3
tqdm>=4.62.0
scikit-learn>=0.24.2
pandas>=1.3.0
transformers>=4.9.2
tensorboard>=2.6.0
wandb>=0.12.0
mamba-ssm>=0.1.0
langdetect>=1.0.9

[dev]
pytest>=6.2.4
black>=21.7b0
flake8>=3.9.2
isort>=5.9.3
mypy>=0.910

[tpu]
torch_xla>=1.9.0
