jax[cpu]
plum-dispatch

[all]
jax[cpu]
plum-dispatch
tensorflow
tensorflow-datasets
datasets
numpy<2.0.0
torch
torchvision
scikit-learn
pandas
nbdev
jupyter
dm-haiku
optax
nbdev-mkdocs

[dev]
jax[cpu]
plum-dispatch
tensorflow
tensorflow-datasets
datasets
numpy<2.0.0
torch
torchvision
scikit-learn
pandas
nbdev
jupyter
dm-haiku
optax
nbdev-mkdocs

[huggingface]
datasets
numpy<2.0.0

[tensorflow]
tensorflow
tensorflow-datasets

[torch]
torch
torchvision
