google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
PythonApache-2.0
Watchers
- akfidjelandDeepMind
- alirezadamashIASBS
- BuddenDAustralia
- CForgie
- cgarciaeGoogle
- CoaLee
- eemailme
- evanatyourserviceDenver, CO
- exbibyteGroq
- fabianp@google-deepmind
- fabioviolaDeepMind
- hbq1London, UK
- jhcloos
- Magnus6803MJ Kontorsservice
- mblondelGoogle DeepMind
- mihaimaruseacGoogle
- mkuneschLondon, UK
- mmhamdyCairo, Egypt
- mtthssLondon, UK
- pharringtonp19
- QuantaMathNone
- rogerpouCatalunya
- rwightman@huggingface
- shantanusharmaSharma Labs
- soraros
- suryabhupa@google-deepmind
- TahianSylvainISPM-Antsobolo
- tomtwostone
- vrouletGoogle
- wangleiphyInstitute of Physics, CAS
- wookayinUniversity of Michigan
- WuShichaoMax Planck Institute for Gravitational Physics
- xmyqsh
- yuzhangbitBeijing