Sea AI Lab

EditAnything

Edit anything in images powered by segment-anything, ControlNet, StableDiffusion, etc. (ACM MM)

envpool

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

Adan

Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models

lorahub

The official repository of paper "LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Composition".

MDT

Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)

metaformer

MetaFormer Baselines for Vision (TPAMI 2024)

CLoT

Official Codebase of our Paper: "Let's Think Outside the Box: Exploring Leap-of-Thought in Large Language Models with Creative Humor Generation" (CVPR 2024)

inceptionnext

InceptionNeXt: When Inception Meets ConvNeXt (CVPR 2024)

zero-bubble-pipeline-parallelism

Zero Bubble Pipeline Parallelism

sewformer

mugs

A PyTorch implementation of Mugs proposed by our paper "Mugs: A Multi-Granular Self-Supervised Learning Framework".

sailor-llm

Sailor: Open Language Models for South-East Asia

Consistent3D

The official PyTorch implementation of Consistent3D (CVPR 2024)

edp

[NeurIPS 2023] Efficient Diffusion Policy

Agent-Smith

Agent Smith: A Single Image Can Jailbreak One Million Multimodal LLM Agents Exponentially Fast

ScaleLong

The official repository of paper "ScaleLong: Towards More Stable Training of Diffusion Model via Scaling Network Long Skip Connection" (NeurIPS 2023)

jax_xc

Exchange correlation functionals translated from libxc to jax

sdft

The official codebase for the paper "Self-Distillation Bridges Distribution Gap in Language Model Fine-tuning".

AnyDoor

AnyDoor: Test-Time Backdoor Attacks on Multimodal Large Language Models

autofd

Automatic Functional Differentiation in JAX

finetune-fair-diffusion

Code of the paper: Finetuning Text-to-Image Diffusion Models for Fairness

d4ft

A JAX library for Density Functional Theory.

MMCBench

DiffMemorize

On Memorization in Diffusion Models

GDPO

Graph Diffusion Policy Optimization

D-TRAK

Intriguing Properties of Data Attribution on Diffusion Models (ICLR 2024)

lm-random-memory-access

Megatron-LLM

distributed trainer for LLMs

zero-bubble-megatron-deepspeed

Zero Bubble Pipeline Parallelism implemented on Megatron-Deepspeed

InfNeRF

InfNeRF: Towards Infinite Scale NeRF Rendering with O(log n) Space Complexity