A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model
MIT License
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
ICLR 2024 论文和开源项目合集
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
Gibbs sampling for generating protein sequences
Adapting LLaMA Decoder to Vision Transformer
[GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregress...
Implementation of Alphafold 3 in Pytorch
Implementation of Soft Actor Critic and some of its improvements in Pytorch
Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in Pytorch
Elucidating the Design Space of Diffusion-Based Generative Models (EDM)
LongLLaMA is a large language model capable of handling long contexts. It is based on OpenLLaMA a...
Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images an...
Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch
Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch