matplotlib
imageio-ffmpeg
imageio
torch
gymnasium
numpy
