keras
scikit-learn
numpy>=1.8
pandas
matplotlib
astropy>=1.3
scipy>=0.14
ipython

[dev]
pytest
pytest-cov
black
flake8
mypy

[gpu-jax]
jax[cuda12]

[gpu-pytorch]
torch[cuda]
torchvision

[gpu-tensorflow]
tensorflow[and-cuda]

[jax]
jax
jaxlib

[pytorch]
torch
torchvision

[tensorflow]
tensorflow
