numpy>=1.19.0
jax>=0.3.10
jaxlib>=0.3.10
opencv-python>=4.2.0.32
matplotlib>=3.3.0
configargparse>=1.2.3
tqdm>=4.50.0
pandas>=1.1.0
scikit-image
scikit-learn
optax>=0.1.3

[gpu]
jax[cuda12]
