optax
Optax is a gradient processing and optimization library for JAX.
What is optax?
Optax is a gradient processing and optimization library for JAX.
Keep exploring
View all alternatives Related tools
100/100
Fabric
Fabric is an open-source framework for augmenting humans using AI. It provides a modular system for solving specific problems using a crowdsourced set of AI pro
100/100
prompt-optimizer
An AI prompt optimizer for writing better prompts and getting better AI results.
100/100
LightGBM
A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classifica