torch>=2.0.0
numpy>=1.21.0
matplotlib>=3.4.3
tqdm>=4.62.0
scikit-learn>=0.24.2
pandas>=1.3.0
transformers>=4.30.0
tensorboard>=2.6.0
wandb>=0.12.0
mamba-ssm>=1.0.0
langdetect>=1.0.9
fastapi>=0.68.0
uvicorn>=0.15.0

[dev]
pytest>=6.2.4
black>=21.7b0
flake8>=3.9.2
isort>=5.9.3
mypy>=0.910
pytest-cov>=2.12.0
pre-commit>=2.15.0

[gpu]
cupy-cuda11x>=11.0.0

[tpu]
torch_xla>=1.9.0
