sleap-io<0.7.0,>=0.6.0
numpy
lightning
kornia
jsonpickle
scipy
attrs
omegaconf
wandb
matplotlib
seaborn
rich
loguru
psutil
opencv-python
hydra-core
jupyter
jupyterlab
pyzmq

[torch]
torch
torchvision>=0.20.0

[torch-cpu]
torch
torchvision>=0.20.0

[torch-cuda118]
torch
torchvision>=0.20.0

[torch-cuda128]
torch
torchvision>=0.20.0

[torch-cuda130]
torch
torchvision>=0.20.0
