optax
There are 0 repository under optax topic.
phlippe / uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
Language:Jupyter Notebook 2133- get-started-with-JAX
gordicaleksa / get-started-with-JAX
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
Language:Jupyter Notebook 557 - EasyDeL
erfanzar / EasyDeL
EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥
Language:Python 148 - diffusion-models-tutorial
varun-ml / diffusion-models-tutorial
Experiment with diffusion models that you can run on your local jupyter instances
Language:Jupyter Notebook 45 hushon / JAX-ResNet-CIFAR10
Simple CIFAR10 ResNet example with JAX.
Language:Python 19- Language:Python 16
MANGA-UOFA / Flora
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors".
Language:Python 12- Language:Python 8
hr0nix / optax-adan
An implementation of adan optimizer for optax
Language:Python 7mzguntalan / h-former
H-Former is a VAE for generating in-between fonts (or combining fonts). Its encoder uses a Point net and transformer to compute a code vector of glyph. Its decoder is composed of multiple independent decoders which act on a code vector to reconstruct a point cloud representing a glpyh.
Language:Python 5qdevpsi3 / quantum-orthogonal-nn
JAX implementation of Classical and Quantum Algorithms for Orthogonal Neural Networks by (Kerenidis et al., 2021)
Language:Python 5ethanluoyc / td3_bc_jax
Direct port of TD3_BC to JAX using Haiku and optax.
Language:Python 4frankroeder / goal_conditioned_rl
Goal-conditioned reinforcement learning like 🔥
Language:Python 4NTT123 / wavernn-16bit
The (unofficial) vanilla version of WaveRNN
Language:Python 4Raffaelbdl / hyperbolic-nn-haiku
dm-haiku implementation of hyperbolic neural networks
Language:Python 4- Language:Python 3
- Language:Python 2
bischtob / Opterax
An gradient-free optimization suite written in JAX. We conform to the optax interface and provide ensemble-based optimizers.
Language:Python 1- nedem
daniel-j-h / nedem
Neural implicit digital elevation model
Language:Python 1 A helper library for training dm-haiku models.
Language:Python 1activatedgeek / optax-swag
Stochastic Weight Averaging (SWA) transforms for Optax with JAX
Language:Python 0BigBalloon8 / Science_Bacc
SQA scottish Science Baccalaureate of Christopher Rae. Improving the preformance of distributed data parallelism of low wifi bandwidths
Language:TeX 0cevenkadir / ferminet_tum
A library which trains the Fermionic Neural Network to find the ground state wave functions of an atom or a molecule using neural network quantum states.
Language:Python 0xghan99 / dsa4212-assignments
This repository contains some of the code I wrote for the assignments in DSA4212 - Optimisation for Large-Scale Data-Driven Inference.
Language:Jupyter Notebook 0alexjackson1 / hk-mnist
An implementation of MNIST classification using LeNet-300-100 in JAX (using Haiku and Optax).
Language:Jupyter NotebookA reimplementation of Parallel DNN Training in JAX by Will Whitney using haiku and optax.
Language:Jupyter Notebook- Language:Jupyter Notebook
- Language:Python
