#pusht
wheel
numpy
pytorch-lightning>=1.7.0,<2.0.0
torch
torchvision
zarr
diffusers
tqdm
gym
pygame
pymunk
shapely
opencv-python
scikit-image
scikit-learn
scikit-video
gdown

# flow_pusht
matplotlib
ipython
termcolor
torchdiffeq
torchsde
torchdyn
torchcfm

# BST quality checks
mako
lxml
terminaltables
pylint>=2.9

# Experiment tracking
wandb
