equinox
jax
jaxtyping
numpy
pillow
optax
torch
torchvision

[demo]
