torch>=2.0
numpy
scipy
xarray

[all]
dcoupler[dev,jax]

[dev]
pytest
pytest-cov

[jax]
jax>=0.4
jaxlib

[models]
