albumentations==1.4.18
av==15.0.0
diffusers==0.35.1
dm-tree==0.1.8
lmdb==1.7.5
msgpack==1.1.0
msgpack-numpy==0.4.8
pandas==2.2.3
peft==0.17.1
termcolor==3.2.0
torch==2.7.0
torchvision==0.22.0
transformers==4.57.0
tyro==0.9.17
flash-attn==2.7.4.post1
click==8.1.8
datasets==3.6.0
einops==0.8.1
gymnasium==1.2.2
matplotlib==3.10.1
numpy==1.26.4
omegaconf==2.3.0
scipy==1.15.3
torchcodec==0.4.0
wandb==0.23.0
pyzmq==27.0.1
deepspeed==0.17.6
moviepy==1.0.3
opencv-contrib-python-headless==4.11.0.86
accelerate>=0.34
safetensors>=0.4.5
huggingface-hub>=0.26
Pillow>=10.0
tqdm>=4.66
requests>=2.32
PyYAML>=6.0
packaging>=24.0
websockets>=13.0

[dev]
ruff
ipython
pytest>=8.0
pytest-timeout>=2.3
hypothesis>=6.0
hypothesis-torch>=0.8

[eval]
imageio>=2.34
draccus>=0.8
transforms3d>=0.4
openpi-client>=0.1

[tensorrt]
onnx>=1.20.0
tensorrt>=10.14.1.48.post1
