numpy
jaxtyping
pybind11
msgpack

[all]
warpconvnet[dev]
warpconvnet[docs]
warpconvnet[models]

[dev]
ipdb
pre-commit
pytest
pytest-benchmark
pytest-cov

[docs]
mkdocs
mkdocs-material
mkdocstrings[python]
pymdown-extensions

[flash-attn]
flash-attn

[models]
rich
h5py
wandb
hydra-core
omegaconf
lightning
torchmetrics

[torch-scatter]
torch-scatter
