Documentation
API
Examples
About
ResNet on CIFAR10 with Flax and JAXopt.
Adversarial training
Dataset distillation
Few-shot Adaptation with Model Agnostic Meta-Learning (MAML)
jax.experimental.pjit example using JAXopt.
jax.pmap example using JAXopt.
Perturbed optimizers