torch
gymnasium
numpy
gymnasium[box2d]
stable_baselines3
