dm-haiku
optax
jax_tqdm
nir

[loaders]
tonic
torchvision==0.19.0+cpu
scikit-learn
