davzha

cruxeval

CRUXEval: Code Reasoning, Understanding, and Execution Evaluation

deep-weight-space-augmentations

Official implementation of "Improved Generalization of Weight Space Networks via Augmentations", ICML 2024

codeit

translation-agent

uspace

An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"

torchtune

PyTorch native finetuning library

zigma

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

neural-graphs

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

recurrent-memory-transformer-pytorch

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

ring-attention-pytorch

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

mamba

Mamba SSM architecture

ModelZooDataset

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

self-guided-diffusion-models

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

fiftyone

Refine high-quality datasets and visual AI models

hyperbolic_learning_library

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

MESH

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

DIFFormer

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

nfn

NF-Layers for constructing neural functionals.

DWSNets

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

ImageBind

ImageBind One Embedding Space to Bind Them All

DIFUSCO

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

torchgfn

GFlowNet library

SEVERE-BENCHMARK

ControlNet

Let us control diffusion models!

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"

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.

I2SB

the-algorithm

Source code for Twitter's Recommendation Algorithm

awesome-flow-matching

A summary of related works about flow matching, stochastic interpolants

paper-qa

High accuracy RAG for answering questions from scientific documents with citations