jax>=0.4.20
jaxlib>=0.4.20
numpy>=1.24.0
xarray>=2023.1.0
netcdf4>=1.6.0
pandas>=2.0.0
optax>=0.1.7
equinox>=0.11.0
typing-extensions>=4.5.0
symfluence>=0.7.0

[all]
jfuse[dev,gpu,viz]

[dev]
pytest>=7.0
pytest-cov>=4.0
black>=23.0
isort>=5.12
mypy>=1.0
ruff>=0.1.0

[gpu]
jax[cuda12_pip]>=0.4.20

[viz]
matplotlib>=3.7.0
geopandas>=0.14.0
contextily>=1.4.0
