datasets
equinox
jax
jaxtyping
numpy
pillow
opencv-python
optax
torch
torchvision

[demo]
