cohen5

catx

🐈‍⬛ Contextual bandits library for continuous action trees with smoothing in JAX

jumanji

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

QDax

Accelerated Quality-Diversity

Mava

🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX

scaling-resnets

⚡️ A framework that investigates the scaling limit of ResNets and compares it to Neural ODEs. Tested on synthetic and standardized datasets. 📈

NeuralCDE

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

progressive_growing_of_GANs

Pure tensorflow implementation of progressive growing of GANs

pytorch-generative-adversarial-networks

A very simple generative adversarial network (GAN) in PyTorch

notebooks