dewml<0.2,>=0.1.0
jax
flax
optax
numpy

[connectome]
pyarrow

[cuda12]
dewml[cuda12]<0.2,>=0.1.0

[cuda13]
dewml[cuda13]<0.2,>=0.1.0

[datasets]
h5py

[nir]
nir

[test]
pytest
ruff==0.14.3
h5py
nir
pyarrow

[tpu]
dewml[tpu]<0.2,>=0.1.0
