gymnasium==1.2.2
stable_baselines3==2.7.0
matplotlib==3.6.0
pydantic==2.12.5
pytest==7.1.1
h5py==3.11.0
datasets==4.4.1
transformers>=4.57.3
accelerate>=0.26.0
torch==2.9.1
torchvision==0.24.1
numpy==1.26.4

[dev]
gymnasium
pytest

[dev:platform_machine != "arm64"]
torch==2.2.2
