# requirements.txt

# Core versions
torch==2.3.0
torchvision==0.18.0
numpy==1.26.4
ipykernel==6.29.3
matplotlib==3.8.3
zarr==2.17.0
hydra-core==1.3.2
einops==0.6.1
tqdm==4.66.4
h5py==3.7.0
gymnasium==0.29.1
pymunk==6.7.0
moviepy==1.0.3
imageio==2.17.0
imageio-ffmpeg==0.4.7
diffusers==0.34.0

# From Git or custom sources
git+https://github.com/facebookresearch/r3m.git
aeon==1.0.0
roma==1.5.2.1
open3d
wandb
pygame
robosuite @ https://github.com/ARISE-Initiative/robosuite/archive/d83a9a76200381c833feda0e06e154c3c01f66bb.tar.gz
robomimic @ https://github.com/ARISE-Initiative/robomimic/archive/8df6e8e482458befaf1decfcd9d635d68a7afcae.tar.gz
