torch
torchvision
wandb
hydra-core
pyyaml
omegaconf
colorama
tqdm
opencv-python
matplotlib
lightning
click
moviepy
imageio
drake
ema_pytorch
accelerate
pytorch_fid
einops
torchmetrics[image]==0.11.4
dm_control
pandas~=1.1
gluonts[torch]==0.13.1
wandb-osh

# decision making
cython<3
d4rl
shimmy>=0.2.1
stable-baselines3
setuptools==65.5.0 
wheel<0.40.0
gym==0.19.0

# for dmlab
internetarchive

# for real robot
pytorchvideo
pyzmq
pyrealsense2
