Skip to content

@sail-sg Sea AI Lab

Change the repository type filter

All

    Repositories list

    100 repositories

    • jrystal

      Public
      A JAX-based Differentiable Density Functional Theory Framework for Materials
      density-functional-theoryquantum-chemistrymaterial-sciencesolid-statejaxdifferentiable-programmingelectron-structure
      Python
      •
      Apache License 2.0
      •1•45•5•2•Updated Feb 15, 2026Feb 15, 2026

    • odc

      Public
      On demand communication
      Python
      •2•32•1•5•Updated Feb 12, 2026Feb 12, 2026

    • Stable-RL

      Public
      Rethinking the Trust Region in LLM Reinforcement Learning
      Python
      •
      Apache License 2.0
      •3•35•0•5•Updated Feb 5, 2026Feb 5, 2026

    • oat

      Public
      🌾 OAT: A research-friendly framework for LLM online alignment, including reinforcement learning, preference learning, etc.
      thompson-samplingalignmentreasoningdistributed-trainingppodueling-banditsdpodistributed-rlllmonline-rl
      Python
      •
      Apache License 2.0
      •59•628•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
      •602•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•182•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.2k•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-differentiationjaxneural-operatorvariational-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
      rlreasoningllmr1-zero
      Python
      •
      MIT License
      •56•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•51•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•73•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-modelattention-mechanismlarge-language-modelsattention-sink
      Python
      •
      MIT License
      •5•155•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-learningoptimizerpytorchartificial-intelligencemoeresnetvitdiffusionmaefairseq
      Python
      •
      Apache License 2.0
      •70•807•5•0•Updated Jun 8, 2025Jun 8, 2025

    • TreeMeshGPT

      Public
      [CVPR 2025] TreeMeshGPT: Artistic Mesh Generation with Autoregressive Tree Sequencing
      Python
      •
      MIT License
      •13•179•5•0•Updated May 22, 2025May 22, 2025

    • ContinualBench

      Public
      Python
      •
      MIT License
      •1•19•1•0•Updated May 20, 2025May 20, 2025

    • zero-bubble-pipeline-parallelism

      Public
      Zero Bubble Pipeline Parallelism
      Python
      •
      Other
      •3.6k•449•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