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: 601
Downloads last week: 3,688
Downloads last month: 14,433