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

[dev]
pytest
pytest-cov
black
pydocstyle
toml
twine
build
ipython
ruff

[docs]
mkdocs
mkdocs-material
mkdocs-jupyter
mike
mkdocstrings[python]
mkdocs-gen-files
mkdocs-literate-nav
mkdocs-section-index

[torch]
torch
torchvision

[torch-cpu]
torch
torchvision

[torch-cuda118]
torch
torchvision

[torch-cuda128]
torch
torchvision
