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.
941 stars
18 watching
67 forks
Language: Python
last commit: about 2 years agoLinked 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,730 |
| google-deepmind/kfac-jax | Library providing an implementation of the K-FAC optimizer and curvature estimator for second-order optimization in neural networks. | 252 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 517 |
 davisyoshida/lorax | A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation. | 134 |
 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. | 932 |
 jycouet/kitql | A collection of standalone tools to speedrun web applications built with GraphQL and Svelte | 409 |
| google-deepmind/jraph | A lightweight library for working with graph neural networks in jax. | 1,380 |
| google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 153 |
 deng-cy/deep_learning_topology_opt | Optimization code for topology optimization problems using machine learning and deep learning techniques | 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 | 797 |
 hyperopt/hyperopt-sklearn | Automates search for optimal parameters in machine learning algorithms. | 1,594 |
 xternalz/sdpoint | A deep learning method for optimizing convolutional neural networks by reducing computational cost while improving regularization and inference efficiency. | 18 |