jaxopt
Deep learning optimizer
An open-source project providing hardware accelerated, batchable and differentiable optimizers in JAX for deep learning.
Hardware accelerated, batchable and differentiable optimizers in JAX.
933 stars
18 watching
66 forks
Language: Python
last commit: 2 months ago
Linked from 1 awesome list
bi-leveldeep-learningdifferentiable-programmingjaxoptimization
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 google-deepmind/optax | 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. | 1,697 |
| google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 248 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 513 |
 davisyoshida/lorax | A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation. | 132 |
 aqibsaeed/genetic-cnn | A tool for exploring and optimizing the architecture of Convolutional Neural Networks using a Genetic Algorithm | 218 |
 darshandeshpande/jax-models | Provides a collection of deep learning models and utilities in JAX/Flax for research purposes. | 151 |
 jankrepl/deepdow | A framework that enables portfolio optimization using deep learning. | 920 |
 jycouet/kitql | A collection of standalone tools to speedrun web applications built with GraphQL and Svelte | 407 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,375 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 152 |
 deng-cy/deep_learning_topology_opt | A toolkit for using machine learning to optimize complex geometries in simulations | 107 |
 deependersingla/deep_portfolio | An algorithm that optimizes portfolio allocation using Reinforcement Learning and Supervised learning. | 168 |
| google-deepmind/chex | A set of utilities for writing reliable JAX code | 788 |
 hyperopt/hyperopt-sklearn | Automates search for optimal parameters in machine learning algorithms. | 1,588 |
 xternalz/sdpoint | A deep learning method for optimizing convolutional neural networks by reducing computational cost while improving regularization and inference efficiency. | 18 |