Terminal Velocity Matching

Making your benchmark of optimization algorithms simple and open

Plots in Julia using the PGFPlots LaTeX package

An extremely fast Python package and project manager, written in Rust.

GitPython is a python library used to interact with Git repositories.

Python 3.8+ toolbox for submitting jobs to Slurm

Flash Attention Triton kernel with support for second-order derivatives

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

Implementation for SE(3) diffusion model with application to protein backbone generation

The Modular Platform (includes MAX & Mojo)

Constrained optimization toolkit for PyTorch

Lambert W function and Lambert W x F distributions in pytorch

gpt-oss-120b and gpt-oss-20b are two open-weight language models by OpenAI

Orbax provides common checkpointing and persistence utilities for JAX users

code for "Residual Flows for Invertible Generative Modeling".

PyTorch Implementation of Diffusion Schrodinger Bridge Matching

Generation of protein sequences and evolutionary alignments via discrete diffusion models

PINDER: The Protein INteraction Dataset and Evaluation Resource

Cell cycle inference in single-cell RNA-seq

A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model

Educational implementation of the Discrete Flow Matching paper

Consistency Models Made Easy

[NeurIPS 2024] Simple and Effective Masked Diffusion Language Model

Beyond Straight-Through

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

Official repo for consistency models.

Minimal Implementation of a D3PM in pytorch