mooninrain

manif

A small C++11 header-only library for Lie theory.

RFdiffusion

Code for running RFdiffusion

llama3

The official Meta Llama 3 GitHub site

lion-pytorch

🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch

schedule_free

Schedule-Free Optimization in PyTorch

score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

k-diffusion

Karras et al. (2022) diffusion models for PyTorch

RMT4ML

Matlab Notebook for visualizing random matrix theory results and their applications to machine learning

RectifiedFlow

Official Implementation of Rectified Flow (ICLR2023 Spotlight)

pre-commit

A framework for managing and maintaining multi-language pre-commit hooks.

ProteinInvBench

The official implementation of the NeurIPS'23 paper ProteinInvBench: Benchmarking Protein Design on Diverse Tasks, Models, and Metrics

Score-Entropy-Discrete-Diffusion

Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)

discrete_flow_models

Code for the paper https://arxiv.org/abs/2402.04997

protein-frame-flow

Fast protein backbone generation with SE(3) flow matching.

tauLDR

Code for the paper https://arxiv.org/abs/2205.14987v2

foldingdiff

Diffusion models of protein structure; trigonometry and attention are all you need!

gin-config

Gin provides a lightweight configuration framework for Python

D3PM-Pytorch

Unofficial PyTorch implementation of Discrete Denoising Diffusion Probabilistic Model(D3PM)

multiflow

OpenDiT

OpenDiT: An Easy, Fast and Memory-Efficient System for DiT Training and Inference

nanotron

Minimalistic large language model 3D-parallelism training

datasets

🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools

mesh-transformer-jax

Model parallel transformers in JAX and Haiku

gpt-neox

An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.

ControlNet

Let us control diffusion models!

protGPT2

Awesome-state-space-models

Collection of papers on state-space models

normalizing-flows

PyTorch implementation of normalizing flow models

awesome-normalizing-flows

Awesome resources on normalizing flows.

ddim

Denoising Diffusion Implicit Models