JAXopt

Documentation

  • Basics
  • Unconstrained optimization
  • Constrained optimization
  • Quadratic programming
  • Non-smooth optimization
  • Stochastic optimization
  • Root finding
  • Fixed point resolution
  • Nonlinear least squares
  • Linear system solving
  • Implicit differentiation
  • Loss and objective functions
  • Line search
  • Perturbed optimization

API

  • API at a glance

Examples

  • Notebook gallery
    • Deep learning
    • Implicit Differentiation
    • Distributed Optimization
    • Perturbed optimizers
  • Example gallery

About

  • Authors
  • Changelog
  • Source code
  • Issue tracker
  • Development
JAXopt
  • »
  • Notebook gallery
  • View page source

Notebook gallery

Deep learning

Resnet example with Flax and JAXopt.

ResNet on CIFAR10 with Flax and JAXopt.

Adversarial Training.

Adversarial training

Implicit Differentiation

Dataset distillation example with JAXopt.

Dataset distillation

Few-shot Adaptation with Model Agnostic Meta-Learning (MAML)

Few-shot Adaptation with Model Agnostic Meta-Learning (MAML)

Distributed Optimization

`jax.experimental.pjit` example using JAXopt.

jax.experimental.pjit example using JAXopt.

`jax.pmap` example using JAXopt.

jax.pmap example using JAXopt.

Perturbed optimizers

Perturbed optimizers with JAXopt.

Perturbed optimizers

Previous Next

© Copyright 2021-2022, the JAXopt authors.

Built with Sphinx using a theme provided by Read the Docs.