Official repo for consistency models.

Morphological snakes for image segmentation and tracking

Physics informed neural network (PINN) for cavity flow governed by Navier-Stokes equation.

Physics Informed Neural Network (PINN) for Burgers' equation.

Implementation of Linformer for Pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning

Tensorflow implementation of "Representation Learning with Contrastive Predictive Coding"

StyleGAN - Official TensorFlow Implementation

Zeolite GAN

A Pytorch Implementation of "Neural Speech Synthesis with Transformer Network"

Code for the paper "Exploration by Random Network Distillation"

Pytorch implementation on OpenAI's Procgen ppo-baseline, built from scratch.

Navier-Stokes fluid simulation in various languages (C, Fortran, and Haskell using Repa)

Code for Go-Explore: a New Approach for Hard-Exploration Problems

[WIP] A colletion of DRL algorithms with intrinsic rewards

Tacotron 2 - PyTorch implementation with faster-than-realtime inference

Examples of published reinforcement learning algorithms in recent literature implemented in TensorFlow