numpy
scipy
matplotlib
pandas
xarray
keras
scikit-learn
netcdf4
pyyaml
tqdm
sphinx
numba
properscoring
pyarrow
imbalanced-learn
bridgescaler
echo-opt
hagelslag
jupyter
nbsphinx
myst_parser
sphinx_book_theme
hagelslag

[jax]
jax

[tensorflow]
tensorflow>=2.16.0

[tensorflow_gpu]
tensorflow[and-cuda]>=2.16.0

[torch]
torch
safetensors
