wandb
ruamel-yaml
configmypy
tensorly
tensorly-torch
torch-harmonics==0.7.3
matplotlib
numpy>=1.25
opt-einsum
h5py
zarr

[all]
pytest
flaky
black
matplotlib
myst-nb
numpydoc
sphinx
sphinx-gallery
tensorly_sphinx_theme
torchtnt
numpy>=1.25
scipy

[dev]
pytest
flaky
black

[doc]
matplotlib
myst-nb
numpydoc
numpy>=1.25
sphinx
sphinx-gallery
tensorly_sphinx_theme
torchtnt
