optax 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 t...

gordicaleksa

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

phlippe

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

tcbrindle

Simple CIFAR10 ResNet example with JAX.

hushon

Accelerate, Optimize performance with streamlined training and serving options with JAX.

erfanzar

Experiment with diffusion models that you can run on your local jupyter instances

varun-ml

JAX implementations of various deep reinforcement learning algorithms.

hamishs

This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.

BorealisAI