numpy
torch
matplotlib
wandb
ale-py
gym
kornia
supersuit
stable_baselines3
einops
