Distributed Shampoo

github.comsite

Implementation of Second Order Optimization Made Practical.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲

   dynamiqs github.com

   High-performance and differentiable simulations of quantum systems with JAX.

   tool topic science link ↗
   
2. ▲

   JAXopt github.com

   Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.

   tool link ↗
   
3. ▲

   QDax github.com

   Quality Diversity optimization in Jax.

   tool link ↗
   
4. ▲

   Quantum Computing Libraries for noncommercial use github.com

   tool topic science link ↗
   
5. ▲

   Distributed github.com

   Distributed computation in Python.

   tool topic programming link ↗
   
6. ▲

   DistributedR github.com

   tool link ↗
   tool

JAX › Models and Projects > Flax: “Implementation of Second Order Optimization Made Practical.”