Optax is a gradient processing and optimization library for JAX.
Repository Index
55,140 visible repositories in the index
1–2 of 2 · Most stars
🎯 A comprehensive gradient-free optimization framework written in Python
- metaheuristics
- optimization
- optimization-algorithms
- python