torch>=2.0.0
numpy>=1.24.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
ruff>=0.1.0
mypy>=1.0.0
gymnasium>=0.29.0

[full]
einops>=0.7.0
gymnasium>=0.29.0
tensorboard>=2.15.0

[transformers]
einops>=0.7.0
