google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
PythonApache-2.0
Stargazers
- bamosMeta AI (FAIR)
- chaoshengtHenan Polytechnic University
- clarkzinzow@eventual
- DevSinghSachanMontreal
- diggerduBytedance AI Lab
- dlsmith
- evanzdShanghai, China
- filangelos@deepmind, @oatml
- fly51flyPRIS
- gdahia
- GitHub30Osaka, Japan
- gmittalPalo Alto, CA
- hansen7@Genentech, @Stanford
- jdmonacoNIH/NINDS
- joaogui1
- kashifBerlin, Germany
- KayhanB21@Voiceflow
- kngwyu@oist
- lartpangDUT
- lidangzzz@Hedgehog-Computing
- madisonmay@IndicoDataSolutions
- makdoudNPalaiseau, France
- mariohsoutoEquilibrium Energy
- markdjwilliams
- mattjjGoogle
- mjsML@NVIDIA
- nixphix@sahajsoft
- ppletscherZurich, Switzerland
- rohit-guptaUniversity of Central Florida
- sherckloNanjing University
- Sreerag-ibtl@InteraktionAB @Vembala
- srihari-humbarwadi@intel
- TezRomacHLimassol, Cyprus
- triwahyuuAichi, Japan
- tshrjnNew York City
- xuyLeaps Capital