psutil>=5.8.0
torch>=1.8.0
GPUtil>=1.4.0

[all]
matplotlib>=3.3.0
numpy
jinja2>=3.0.0
transformers>=4.0.0

[dev]
pytest>=6.0
flake8>=3.8
black>=21.0b0
coverage>=5.0
mypy>=0.900
ipykernel

[html]
jinja2>=3.0.0

[plotting]
matplotlib>=3.3.0
numpy

[trl]
transformers>=4.0.0
