TUPE

parti-pytorch

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

sentiment-discovery

Unsupervised Language Modeling at scale for robust sentiment classification

iTransformer

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

SimCR

Code for NAACL 2024 main conference paper "An Empirical Study of Consistency Regularization for E...

MEGABYTE-pytorch

Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Py...

Robust-Prefix-Tuning

code for the ICLR'22 paper: On Robust Prefix-Tuning for Text Classification

BertSum

Code for paper Fine-tune BERT for Extractive Summarization

clip-jax

Train vision models using JAX and 🤗 transformers

ProteinDT

GLM

GLM (General Language Model)

recurrent-memory-transformer-pytorch

Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch

chain-of-hindsight

Chain-of-Hindsight, A Scalable RLHF Method

x-clip

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

progen

Implementation and replication of ProGen, Language Modeling for Protein Generation, in Jax

audiolm-pytorch

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