clip-jax

imagen-pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

PaLM-rlhf-pytorch

Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architectu...

soundstorm-pytorch

Implementation of SoundStorm, Efficient Parallel Audio Generation from Google Deepmind, in Pytorch

DALLE2-pytorch

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

prisma

Computational photography pipeline that performs multiple inferences from any image or video.

alphafold3-pytorch

Implementation of Alphafold 3 in Pytorch

transformers

DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

muse-maskgit-pytorch

Implementation of Muse: Text-to-Image Generation via Masked Generative Transformers, in Pytorch

voicebox-pytorch

Implementation of Voicebox, new SOTA Text-to-speech network from MetaAI, in Pytorch

audiolm-pytorch

Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Re...

iTransformer

Unofficial implementation of iTransformer - SOTA Time Series Forecasting using Attention networks...

phenaki-pytorch

Implementation of Phenaki Video, which uses Mask GIT to produce text guided videos of up to 2 min...

parti-pytorch

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

x-clip

A concise but complete implementation of CLIP with various experimental improvements from recent ...