davzha

the-algorithm

Source code for Twitter's Recommendation Algorithm

segment-anything

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

ControlNet

Let us control diffusion models!

mamba

Mamba SSM architecture

xformers

Hackable and optimized Transformers building blocks, supporting a composable construction.

ImageBind

ImageBind One Embedding Space to Bind Them All

fiftyone

The open-source tool for building high-quality datasets and computer vision models

paper-qa

LLM Chain for answering questions from documents with citations

DiffusionDet

[ICCV2023 Best Paper Finalist] PyTorch implementation of DiffusionDet (https://arxiv.org/abs/2211.09788)

FollowYourPose

[AAAI 2024] Follow-Your-Pose: This repo is the official implementation of "Follow-Your-Pose : Pose-Guided Text-to-Video Generation using Pose-Free Videos"

conditional-flow-matching

TorchCFM: a Conditional Flow Matching library

Adan

Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models

STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences

ring-attention-pytorch

Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch

recurrent-memory-transformer-pytorch

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

Awesome-GFlowNets

A curated list of resources about generative flow networks (GFlowNets).

DIFFormer

The official implementation for ICLR23 spotlight paper "DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion"

I2SB

zigma

A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)

awesome-flow-matching

A summary of related works about flow matching, stochastic interpolants

torchgfn

GFlowNet library

DIFUSCO

Code of NeurIPS paper: arxiv.org/abs/2302.08224

hyperbolic_learning_library

An extension of the PyTorch library containing various tools for performing deep learning in hyperbolic space.

DWSNets

Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]

nfn

NF-Layers for constructing neural functionals.

neural-graphs

Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).

ModelZooDataset

Model Zoos published at the NeurIPS 2022 Dataset & Benchmark track: "Model Zoos: A Dataset of Diverse Populations of Neural Network Models"

SEVERE-BENCHMARK

self-guided-diffusion-models

An official implementation of CVPR 2023 "Self-Guided Diffusion Models"

MESH

Code for the paper "Unlocking Slot Attention by Changing Optimal Transport Costs"