Extending JAX with custom C++ and CUDA code

High-Resolution Image Synthesis with Latent Diffusion Models

The most accessible way of tinkering with a GPT - one hackable script

Examples of how to create colorful, annotated equations in Latex using Tikz.

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 a…

Matplotlib styles for scientific plotting

"Probabilistic Machine Learning" - a book series by Kevin Murphy

Just playing with getting VQGAN+CLIP running locally, rather than having to use colab.

RL implementations

A collection of useful .gitignore templates

Softlearning is a reinforcement learning framework for training maximum entropy policies in continuous domains. Includes the official implementation of the Soft Actor-Critic algorithm.

Official implementation of DynE, Dynamics-aware Embeddings for RL

Jax (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

Open world survival environment for reinforcement learning

Code to support training, evaluating and interacting neural network dialog models, and training them with reinforcement learning. Code to deploy a web server which hosts the models live online is a…

Code for the paper "Phasic Policy Gradient"

Trax — Deep Learning with Clear Code and Speed

Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Minimalistic gridworld package for OpenAI Gym

High throughput asynchronous reinforcement learning

Repository for the paper "Planning to Explore via Self-Supervised World Models"

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

Dream to Control: Learning Behaviors by Latent Imagination

DrQ: Data regularized Q