lilisierrayu

gcp

triton

Development repository for the Triton language and compiler

metaseq-opt

Repo for external large-scale work

lm-evaluation-harness

A framework for few-shot evaluation of autoregressive language models.

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

mesh-transformer-jax

Model parallel transformers in JAX and Haiku

latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models

taming-transformers

Taming Transformers for High-Resolution Image Synthesis

transformer-in-transformer

Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorch