absl-py
datasets
drawdata
editdistance
einops
matplotlib
numpy
pyyaml
schedulefree
scikit-learn
scipy
simplipy
torch
torch_optimizer
tqdm
wandb
zss

[dev]
pre-commit
pytest
pytest-cov
mypy
flake8
pygount
pylint
types-setuptools
types-tqdm
types-toml
types-PyYAML
radon
mkdocs
mkdocs-material
mkdocs-autorefs
mkdocstrings
mkdocs-get-deps
mkdocs-material-extensions
mkdocstrings-python
