luyifanlu

External-Attention-pytorch

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

DALLE2-pytorch

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models

trlx

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

marqo

Unified embedding generation and search engine. Also available on cloud - cloud.marqo.ai

CogVideo

Text-to-video generation. The repo for ICLR2023 paper "CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformers"

prompt-to-prompt

alphatensor

diplomacy_cicero

Code for Cicero, an AI agent that plays the game of Diplomacy with open-domain natural language negotiation.

Cold-Diffusion-Models

Official implementation of Cold-Diffusion for different transformations in pytorch.

ProteinMPNN

Code for the ProteinMPNN paper

Poisson_flow

Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)

Adan

Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models

ts2vec

A universal time series representation learning framework

EquiBind

EquiBind: geometric deep learning for fast predictions of the 3D structure in which a small molecule binds to a protein

git-re-basin

Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"

HorNet

[NeurIPS 2022] HorNet: Efficient High-Order Spatial Interactions with Recursive Gated Convolutions

Flowformer

About Code release for "Flowformer: Linearizing Transformers with Conservation Flows" (ICML 2022), https://arxiv.org/pdf/2202.06258.pdf

ICML2022-FEDformer

Source code of ICML'22 paper: FEDformer: Frequency Enhanced Decomposed Transformer for Long-term Series Forecasting

Rethinking-Anomaly-Detection

"Rethinking Graph Neural Networks for Anomaly Detection" in ICML 2022

FedTree

A tree-based federated learning system (MLSys 2023)

language_modeling_via_stochastic_processes

Language modeling via stochastic processes. Oral @ ICLR 2022.

InferOpt.jl

Combinatorial optimization layers for machine learning pipelines

reStructured-Pretraining

reStructured Pre-training

MODEST

Code release for "Learning to Detect Mobile Objects from LiDAR Scans Without Labels" [CVPR 2022]

mengzi-zero-shot

NLU & NLG (zero-shot) depend on mengzi-t5-base-mt pretrained model

SynNet

debug-mistakes-cce

Meaningfully debugging model mistakes with conceptual counterfactual explanations. ICML 2022

git-re-basin-pytorch

Git Re-Basin: Merging Models modulo Permutation Symmetries in PyTorch

deepnash