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

[acceleration]
accelerate>=0.20.0

[all]
pytest>=7.0.0
black>=22.0.0
isort>=5.10.0
mypy>=0.991
psutil>=5.9.0
GPUtil>=1.4.0
tqdm>=4.64.0
matplotlib>=3.5.0
seaborn>=0.11.0
jupyter>=1.0.0
datasets>=2.0.0
accelerate>=0.20.0

[dev]
pytest>=7.0.0
black>=22.0.0
isort>=5.10.0
mypy>=0.991

[examples]
tqdm>=4.64.0
matplotlib>=3.5.0
seaborn>=0.11.0
jupyter>=1.0.0
datasets>=2.0.0

[monitoring]
psutil>=5.9.0
GPUtil>=1.4.0
