Download
Optax is a gradient processing and optimization library for JAX. It is designed to facilitate research by providing building blocks that can be recombined in custom ways in order to optimize parametric models such as, but not limited to, deep neural networks. We favor focusing on small composable building blocks that can be effectively combined into custom solutions. Others may build upon these basic components in more complicated abstractions. Whenever reasonable, implementations prioritize readability and structuring code to match standard equations, over code reuse.
Features
- Optax is a gradient processing and optimization library for JAX
- Provides simple, well-tested, efficient implementations of core components
- Documentation available
- Examples available
- Improve research productivity by enabling to easily combine low-level ingredients into custom optimisers (or other gradient processing components)
- Accelerate adoption of new ideas by making it easy for anyone to contribute
- You can install the latest released version of Optax from PyPI
Project Samples
Categories
Machine LearningLicense
Apache License V2.0Follow Optax
Other Useful Business Software
PRTG Catches Network Issues Before They Cause Downtime
Reactive troubleshooting usually means hearing about a problem from frustrated users, not your monitoring tool. PRTG sets threshold-based alerts across devices, servers and applications, notifying your team by email, SMS or push the moment a metric crosses a set limit. That means catching a failing disk or overloaded server before it becomes an outage and getting time back from firefighting. Start a free trial and set your first alerts today.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of Optax!
