torch>=2.1
numpy>=1.23
einops>=0.7
tqdm>=4.66
pyyaml>=6.0

[all]
pyjepa[data,planning,train,video,vision]

[data]
h5py>=3.10
datasets>=2.16

[dev]
pytest>=7.4
ruff>=0.4
black>=24.0

[planning]
scipy>=1.11

[train]
wandb>=0.16
tensorboard>=2.15

[video]
torchvision>=0.16
av>=11.0

[video:platform_system != "Darwin"]
decord>=0.6

[vision]
torchvision>=0.16
Pillow>=10.0
