A simple but complete full-attention transformer
Implementation of Imagen, Google's Text-to-Image Neural Network
Making large AI models cheaper, faster and more accessible
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
A simple forecasting package
Build cross-modal and multimodal applications on the cloud
Fast ODE Solver for Diffusion Probabilistic Model Sampling
A fork of OpenAI Baselines, implementations of reinforcement learning
Identification codes
Deep Learning GPU training system