Code for "What really matters in matrix-whitening optimizers?"

A set of TFDS dataset builders for common datasets

Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...

Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions

Flow-matching algorithms in JAX

Implementation of Diffusion Transformer (DiT) in JAX

JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)

Frechet inception distance (FID) evaluation in JAX

Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"

Brief Exploration of Generative Models

code for CLIPDraw

Code for "Selecting for Selection"

code for "Quality Diversity: Evolving Ocean Creatures"

type for the AI

count your youtube views

a base for building Boost to link C++ with python

generate MNIST using a Variational Autoencoder

random search, hill climbing, policy gradient

a DRAW model for colored images in tensorflow