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

MIDI / symbolic music tokenizers for Deep Learning models 🎶

Open-source Auth0/Clerk alternative

PyTorch interface for TrueGrad Optimizers

Efficient PScan implementation in PyTorch

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Game Boy emulator written in Python

Playing Pokemon Red with Reinforcement Learning

Ungreedy subword tokenizer and vocabulary trainer for Python, Go & Javascript

ComfyUI nodes based on the paper "FABRIC: Personalizing Diffusion Models with Iterative Feedback" (Feedback via Attention-Based Reference Image Conditioning)

A concise but complete full-attention transformer with a set of promising experimental features from various papers

A python framework for symbolic music generation, evaluation and analysis

Speed up Stable Diffusion with this one simple trick!

Better Aligning Text-to-Image Models with Human Preference. ICCV 2023

Audio generation using diffusion models, in PyTorch.

Implementation of Parti, Google's pure attention-based text-to-image neural network, in Pytorch

Source code for "FIGARO: Generating Symbolic Music with Fine-Grained Artistic Control"

Simple, single-file fluid solvers for learning purposes

Client/server side PDF printing in pure JavaScript