numpy
scipy
torch>=1.8.0
torchvision

[dev]
pytest
pytest-cov
matplotlib
requests
astropy
tensorboard
sphinx>=2.3.0
numpy
jupytext
nbsphinx
faculty-sphinx-theme
sphinx_copybutton
jupyter
nbconvert
matplotlib
sphinxcontrib-mermaid
astropy
tensorboard
pylint
black
pre-commit

[docs]
sphinx>=2.3.0
numpy
jupytext
nbsphinx
faculty-sphinx-theme
sphinx_copybutton
jupyter
nbconvert
matplotlib
sphinxcontrib-mermaid
astropy
tensorboard

[test]
pytest
pytest-cov
matplotlib
requests
astropy
tensorboard
