jaxopt
by google
Hardware accelerated, batchable and differentiable optimizers in JAX.
AI summary
Deep learning optimizer
An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning.
- stars
- 941
- forks
- 67
- watching
- 18
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Gradient optimizer library
A gradient processing and optimization library designed to facilitate research and productivity in machine learning by providing building blocks for custom optimizers and gradient processing components.
KFAC optimizer
Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks.
Optimization layer
A PyTorch module that adds differentiable optimization as a layer to neural networks
Memory optimizer
A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation.
CNN optimizer
A tool for exploring and optimizing the architecture of Convolutional Neural Networks using a Genetic Algorithm
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
Portfolio optimizer
A framework that enables portfolio optimization using deep learning.
Web app optimizer
A collection of standalone tools to speedrun web applications built with GraphQL and Svelte
Graph library
A lightweight library for working with graph neural networks in jax.
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Topology optimizer
Optimization code for topology optimization problems using machine learning and deep learning techniques
Portfolio optimizer
An algorithm that optimizes portfolio allocation using Reinforcement Learning and Supervised learning.
JAX helper library
A set of utilities for writing reliable JAX code
parameter optimizer
Automates search for optimal parameters in machine learning algorithms.
Inference optimizer
A deep learning method for optimizing convolutional neural networks by reducing computational cost while improving regularization and inference efficiency.
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jaxopt.
