mlagents==1.0.0
stable-baselines3[extra]==1.8.0
sb3-contrib==1.8.0
torchvision
timm
nvidia-ml-py

[notebook]
ipywidgets
