requirements.txt

captum==0.6.0
gymnasium==0.29.1
gymnasium[box2d]
matplotlib==3.8.0
matplotlib-inline==0.1.6
torch==2.1.0
numpy==1.26.0
mujoco==2.3.7
moviepy==1.0.3
imageio==2.31.5
imageio-ffmpeg==0.4.9
AutoROM==0.4.2
scipy
AutoROM.accept-rom-license==0.6.1
ale-py==0.8.1