Sea AI Lab

All

100 repositories

Stable-RL
Public
Rethinking the Trust Region in LLM Reinforcement Learning
Python
•
Apache License 2.0
•4•39•0•5•Updated Mar 2, 2026Mar 2, 2026
odc
Public
On demand communication
Python
•2•32•1•3•Updated Feb 26, 2026Feb 26, 2026
jrystal
Public
A JAX-based Differentiable Density Functional Theory Framework for Materials
density-functional-theory quantum-chemistry material-science solid-state jax differentiable-programming electron-structure
Python
•
Apache License 2.0
•1•45•5•2•Updated Feb 15, 2026Feb 15, 2026
oat
Public
🌾 OAT: A research-friendly framework for LLM online alignment, including reinforcement learning, preference learning, etc.
thompson-sampling alignment reasoning distributed-training ppo dueling-bandits dpo distributed-rl llm online-rl
Python
•
Apache License 2.0
•59•633•6•1•Updated Jan 29, 2026Jan 29, 2026
LifelongSafetyAlignment
Public
Python
•0•11•1•0•Updated Jan 13, 2026Jan 13, 2026
feedback-conditional-policy
Public
Code for "Language Models Can Learn from Verbal Feedback Without Scalar Rewards"
Python
•2•59•0•0•Updated Jan 5, 2026Jan 5, 2026
InfNeRF
Public
InfNeRF: Towards Infinite Scale NeRF Rendering with O(log n) Space Complexity
Python
•
Apache License 2.0
•1•12•1•0•Updated Jan 3, 2026Jan 3, 2026
SkyLadder
Public
The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling
Python
•
Apache License 2.0
•601•42•1•0•Updated Dec 29, 2025Dec 29, 2025
d4ft
Public
A JAX library for Density Functional Theory.
Python
•
Apache License 2.0
•5•54•16•0•Updated Nov 25, 2025Nov 25, 2025
Precision-RL
Public
Defeating the Training-Inference Mismatch via FP16
Python
•
MIT License
•15•183•4•0•Updated Nov 14, 2025Nov 14, 2025
Precision-RL-verl
Public
Defeating the Training-Inference Mismatch via FP16
Python
•
Apache License 2.0
•3.3k•5•0•0•Updated Nov 14, 2025Nov 14, 2025
NDA
Public
Code for "Nonparametric Data Attribution for Diffusion Models"
Jupyter Notebook
•0•15•1•0•Updated Nov 11, 2025Nov 11, 2025
tty-use
Public
C
•0•14•0•0•Updated Oct 13, 2025Oct 13, 2025
imperceptible-jailbreaks
Public
[ArXiv 2025] Imperceptible Jailbreaking against Large Language Models
Python
•5•24•0•0•Updated Oct 7, 2025Oct 7, 2025
variational-reasoning
Public
Code for "Variational Reasoning for Language Models"
Python
•1•56•1•0•Updated Sep 29, 2025Sep 29, 2025
autofd
Public
Automatic Functional Differentiation in JAX
automatic-differentiation jax neural-operator variational-calculus
Python
•
Apache License 2.0
•1•81•6•0•Updated Sep 18, 2025Sep 18, 2025
BanditSpec
Public
Python
•2•5•0•0•Updated Sep 2, 2025Sep 2, 2025
understand-r1-zero
Public
Understanding R1-Zero-Like Training: A Critical Perspective
rl reasoning llm r1-zero
Python
•
MIT License
•57•1.2k•9•0•Updated Aug 27, 2025Aug 27, 2025
Video-Next-Event-Prediction
Public
Python
•
MIT License
•1•20•3•0•Updated Aug 9, 2025Aug 9, 2025
AnytimeReasoner
Public
Optimizing Anytime Reasoning via Budget Relative Policy Optimization
Python
•
Apache License 2.0
•3•52•0•0•Updated Jul 15, 2025Jul 15, 2025
LongSpec
Public
LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification
Python
•
MIT License
•3•74•0•0•Updated Jul 14, 2025Jul 14, 2025
Attention-Sink
Public
[ICLR 2025] When Attention Sink Emerges in Language Models: An Empirical View (Spotlight)
language-model attention-mechanism large-language-models attention-sink
Python
•
MIT License
•5•157•1•0•Updated Jul 8, 2025Jul 8, 2025
VeriFree
Public
Reinforcing General Reasoning without Verifiers
Python
•6•96•7•0•Updated Jun 24, 2025Jun 24, 2025
Adan
Public
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
deep-learning optimizer pytorch artificial-intelligence moe resnet vit diffusion mae fairseq
Python
•
Apache License 2.0
•71•808•5•0•Updated Jun 8, 2025Jun 8, 2025
TreeMeshGPT
Public
[CVPR 2025] TreeMeshGPT: Artistic Mesh Generation with Autoregressive Tree Sequencing
Python
•
MIT License
•14•180•5•0•Updated May 22, 2025May 22, 2025
ContinualBench
Public
Python
•
MIT License
•1•19•2•0•Updated May 20, 2025May 20, 2025
zero-bubble-pipeline-parallelism
Public
Zero Bubble Pipeline Parallelism
Python
•
Other
•3.6k•451•29•0•Updated May 7, 2025May 7, 2025
FlowReasoner
Public
Python
•7•144•1•0•Updated May 6, 2025May 6, 2025
Meta-Unlearning
Public
Python
•2•33•1•0•Updated Apr 22, 2025Apr 22, 2025
LightTrans
Public
The official implementation of "LightTransfer: Your Long-Context LLM is Secretly a Hybrid Model with Effortless Adaptation"
Python
•0•22•0•0•Updated Apr 22, 2025Apr 22, 2025