Sea AI Lab

All

93 repositories

feedback-conditional-policy
Public
Python
•0•3•0•0•Updated Sep 26, 2025Sep 26, 2025
variational-reasoning
Public
0•3•0•0•Updated Sep 25, 2025Sep 25, 2025
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
•2•35•5•0•Updated Sep 25, 2025Sep 25, 2025
LifelongSafetyAlignment
Public
Python
•0•10•1•0•Updated Sep 20, 2025Sep 20, 2025
autofd
Public
Automatic Functional Differentiation in JAX
automatic-differentiation jax neural-operator variational-calculus
Python
•
Apache License 2.0
•1•77•6•0•Updated Sep 18, 2025Sep 18, 2025
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
•32•469•1•1•Updated Sep 15, 2025Sep 15, 2025
BanditSpec
Public
Python
•0•0•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
•51•1.1k•7•0•Updated Aug 27, 2025Aug 27, 2025
SkyLadder
Public
The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling
Python
•
Apache License 2.0
•569•34•0•0•Updated Aug 25, 2025Aug 25, 2025
Video-Next-Event-Prediction
Public
Python
•
MIT License
•1•19•3•0•Updated Aug 9, 2025Aug 9, 2025
AnytimeReasoner
Public
Optimizing Anytime Reasoning via Budget Relative Policy Optimization
Python
•
Apache License 2.0
•2•48•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•63•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•125•1•0•Updated Jul 8, 2025Jul 8, 2025
VeriFree
Public
Reinforcing General Reasoning without Verifiers
Python
•7•86•7•0•Updated Jun 24, 2025Jun 24, 2025
d4ft
Public
A JAX library for Density Functional Theory.
Python
•
Apache License 2.0
•5•54•16•0•Updated Jun 18, 2025Jun 18, 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
•69•798•3•0•Updated Jun 8, 2025Jun 8, 2025
TreeMeshGPT
Public
[CVPR 2025] TreeMeshGPT: Artistic Mesh Generation with Autoregressive Tree Sequencing
Python
•
MIT License
•9•165•3•0•Updated May 22, 2025May 22, 2025
ContinualBench
Public
Python
•
MIT License
•0•11•1•0•Updated May 20, 2025May 20, 2025
zero-bubble-pipeline-parallelism
Public
Zero Bubble Pipeline Parallelism
Python
•
Other
•3.1k•427•29•0•Updated May 7, 2025May 7, 2025
FlowReasoner
Public
Python
•7•134•1•0•Updated May 6, 2025May 6, 2025
Meta-Unlearning
Public
Python
•1•28•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•20•0•0•Updated Apr 22, 2025Apr 22, 2025
ActivePRM
Public
Jupyter Notebook
•0•18•0•0•Updated Apr 16, 2025Apr 16, 2025
dice
Public
Official implementation of Bootstrapping Language Models via DPO Implicit Rewards
alignment preference-learning large-language-models rlhf
Python
•
MIT License
•3•44•0•0•Updated Apr 15, 2025Apr 15, 2025
oat-zero
Public
A lightweight reproduction of DeepSeek-R1-Zero with indepth analysis of self-reflection behavior.
Python
•
MIT License
•10•244•0•0•Updated Apr 15, 2025Apr 15, 2025
jax_xc
Public
Exchange correlation functionals translated from libxc to jax
python density-functional-theory libxc jax
Python
•
Mozilla Public License 2.0
•2•45•6•0•Updated Mar 24, 2025Mar 24, 2025
CPO
Public
[NeurIPS 2024] The official implementation of paper: Chain of Preference Optimization: Improving Chain-of-Thought Reasoning in LLMs.
Python
•8•128•0•0•Updated Mar 21, 2025Mar 21, 2025
sailor2
Public
🔱 Sailor2: Sailing in South-East Asia with Inclusive Multilingual LLMs
vietnamese indonesia thai tamil tagalog cebuano language-model burmese khmer lao
3•67•0•0•Updated Mar 21, 2025Mar 21, 2025
Megatron-Sailor2
Public
Megatron for Sailor2/Qwen2.5
Python
•2•4•0•0•Updated Mar 21, 2025Mar 21, 2025
AR-Video-Diffusion
Public
JavaScript
•0•0•0•0•Updated Mar 17, 2025Mar 17, 2025