Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2021/Spring 2022
-
Updated
Jun 27, 2022 - Jupyter Notebook
#
optax
Here are 13 public repositories matching this topic...
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.
-
Updated
Jan 22, 2022 - Jupyter Notebook
JAX implementations of various deep reinforcement learning algorithms.
-
Updated
Apr 23, 2021 - Python
The (unofficial) vanilla version of WaveRNN
-
Updated
Dec 7, 2021 - Python
JAX/Flax implementation of finite-size scaling
-
Updated
May 5, 2022 - Python
Direct port of TD3_BC to JAX using Haiku and optax.
-
Updated
Sep 7, 2021 - Python
Neural implicit digital elevation model
-
Updated
May 25, 2022 - Python
An gradient-free optimization suite written in JAX. We conform to the optax interface and provide ensemble-based optimizers.
-
Updated
Dec 8, 2021 - Python
JAX implementation of Classical and Quantum Algorithms for Orthogonal Neural Networks by (Kerenidis et al., 2021)
quantum-computing
mnist-classification
haiku
stax
jax
optax
quantum-inspired
orthogonal-neural-network
-
Updated
Dec 7, 2021 - Python
A helper library for training dm-haiku models.
-
Updated
Feb 7, 2021 - Python
Simple JAX implementation of Gaussian process regression
-
Updated
Feb 12, 2022
Improve this page
Add a description, image, and links to the optax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the optax topic, visit your repo's landing page and select "manage topics."