google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
PythonApache-2.0
Stargazers
- agibsonccchttp://konduit.ai/
- ahmgeek@gitlabhq
- alokBerkeley, CA
- d-behl
- EddieCunningham
- EdgeLLMRenmin University of China
- erwtokritosAthens, Greece
- fanyang01@apecloud
- fd0r@zama-ai
- fdasilva59Scaleway
- gd-zhangUniversity of Toronto
- haraballOffenbach am Main, Germany
- himktTokyo, Japan
- iamaadityaML Researcher
- jvalinsky
- KeAWangStanford, CA
- lmazzonItaltel SpA
- maxentileRelay Therapeutics
- minaminaoTokyo
- mohammedaugie13@bitwyre
- Nicolaus93
- nikitavoloboevMadrid
- OriolAbrilBarcelona
- parasjUC Berkeley
- rasmi@google-org
- rish-16@KrishnaswamyLab @Graph-and-Geometric-Learning
- rohan-paulhttps://www.linkedin.com/in/rohan-paul-ai
- sdan
- ShayekhBinIslamDhaka, Bangladesh
- Siddharth-Shrivastava7IIT Hyderabad
- simonbatznerHarvard
- singhranjodhIndia
- TheShadow29Meta
- thiagopbueno@Luizalabs
- timlautkPhiladelphia, PA
- TuanNguyen27Salesforce Einstein