optax-shampoo

PyPI page
Home page
Author: rohananil
License: Apache Software License
Summary: Distributed Shampoo (Second order Optimizer for Deep Learning) Optax Optimizer
Latest version: 0.0.6


Downloads last day: 721
Downloads last week: 5,846
Downloads last month: 26,064