dm-haiku
optax
jax_tqdm
nir

[data]
tonic
torchvision
sklearn
