torch>=1.9.0
transformers>=4.20.0
numpy>=1.21.0

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=22.0
isort>=5.0
flake8>=4.0
mypy>=0.910

[examples]
jupyter>=1.0.0
notebook>=6.0.0
ipywidgets>=7.0.0
trl>=0.7.0
unsloth

[wandb]
wandb>=0.12.0
