torch
geoopt>=0.5.1
numpy
tqdm
cvxpy
scikit-learn==1.5.1
datasets

[dev]
mypy
jaxtyping
beartype
pytest
pytest-cov
pytest-env
ruff
pre-commit
typing_extensions
pygments

[docs]
mkdocs
mkdocs-material
mkdocstrings[python]
mkdocs-autorefs
mkdocs-gen-files
griffe
