skeletondyh

Reflected-Diffusion

[ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)

GRED

[ICML 2024] Recurrent Distance Filtering for Graph Representation Learning

FoldFlow

FoldFlow: SE(3)-Stochastic Flow Matching for Protein Backbone Generation

riemannian-fm

code for "Riemannian Flow Matching on General Geometries".

al-folio

A beautiful, simple, clean, and responsive Jekyll theme for academics

geometric_ml

This repository contains code for applying Riemannian geometry in machine learning.

alphaflow

AlphaFold Meets Flow Matching for Generating Protein Ensembles

chroma

A generative model for programmable protein design

e3nn

A modular framework for neural networks with Euclidean symmetry

open-interpreter

A natural language interface for computers

consistency_models

Official repo for consistency models.

attention-rank-collapse

[ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.

llama

Inference code for Llama models

tuning_playbook

A playbook for systematically maximizing the performance of deep learning models.

Awesome-Diffusion-Models

A collection of resources and papers on Diffusion Models

X-Decoder

[CVPR 2023] Official Implementation of X-Decoder for generalized decoding for pixel, image and language

awesome-jax

JAX - A curated list of resources https://github.com/google/jax

lrgb

Long Range Graph Benchmark, NeurIPS 2022 Track on D&B

in-context-learning

cords

Reduce end to end training time from days to hours (or hours to minutes), and energy requirements/costs by an order of magnitude using coresets and data selection.

jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

stable-diffusion

A latent text-to-image diffusion model

TransformersCanDoBayesianInference

Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper

how-do-vits-work

(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"

SAT

Official Pytorch code for Structure-Aware Transformer.

DatasetCondensation

Dataset Condensation (ICLR21 and ICML21)

syne-tune

Large scale and asynchronous Hyperparameter and Architecture Optimization at your fingertips.

unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

s4

Structured state space sequence models

diffstride

TF/Keras code for DiffStride, a pooling layer with learnable strides.