sleap-io<0.7.0,>=0.6.5
numpy
torch
torchvision>=0.20.0
lightning
skia-python<=138.0,>=87.0
jsonpickle
scipy
attrs
omegaconf
wandb
matplotlib
seaborn
rich
textual>=0.40.0
loguru
psutil
opencv-python
hydra-core
jupyter
jupyterlab
pyzmq
rich-click>=1.9.5

[cpu]
sleap-nn[torch-cpu]

[export]
onnx>=1.15.0
onnxruntime>=1.16.0
onnxscript>=0.1.0

[export-gpu]
onnx>=1.15.0
onnxruntime-gpu>=1.16.0
onnxscript>=0.1.0

[gpu]
sleap-nn[torch-cuda130]

[tensorrt]

[tensorrt:sys_platform == "linux" or sys_platform == "win32"]
tensorrt>=10.13.0
torch-tensorrt>=2.5.0

[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
