torch>=2.0.0
numpy>=1.20.0
scipy>=1.7.0
matplotlib>=3.5.0
tqdm>=4.60.0
scikit-learn>=1.0.0

[all]
transformers>=4.30.0
torchvision>=0.15.0
gymnasium>=0.26.0
stable-baselines3>=2.0.0
pytest>=7.0.0

[test]
pytest>=7.0.0
