Repositories list

• jrystal

  Public
  A JAX-based Differentiable Density Functional Theory Framework for Materials

  density-functional-theoryquantum-chemistrymaterial-science+ 4solid-statejaxdifferentiable-programmingelectron-structure

  Python

  •

  Apache License 2.0

  •1•14•6•3•Updated May 20, 2025May 20, 2025

• AnytimeReasoner

  Public
  Optimizing Anytime Reasoning via Budget Relative Policy Optimization

  Python

  •

  Apache License 2.0

  •0•9•0•0•Updated May 20, 2025May 20, 2025

• d4ft

  Public
  A JAX library for Density Functional Theory.

  Python

  •

  Apache License 2.0

  •5•53•16•0•Updated May 20, 2025May 20, 2025

• ContinualBench

  Public
  Python

  •

  MIT License

  •0•3•0•0•Updated May 20, 2025May 20, 2025

• oat

  Public
  🌾 OAT: A research-friendly framework for LLM online alignment, including preference learning, reinforcement learning, etc.

  thompson-samplingalignmentreasoning+ 13distributed-trainingppodueling-banditsdpodistributed-rlllmonline-rl+ 6

  Python

  •

  Apache License 2.0

  •26•356•4•2•Updated May 19, 2025May 19, 2025

• zero-bubble-pipeline-parallelism

  Public
  Zero Bubble Pipeline Parallelism

  Python

  •

  Other

  •2.8k•394•28•0•Updated May 7, 2025May 7, 2025

• FlowReasoner

  Public
  Python

  •6•108•0•0•Updated May 6, 2025May 6, 2025

• Meta-Unlearning

  Public
  Python

  •1•27•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•19•0•0•Updated Apr 22, 2025Apr 22, 2025

• TreeMeshGPT

  Public
  [CVPR 2025] TreeMeshGPT: Artistic Mesh Generation with Autoregressive Tree Sequencing

  Python

  •

  MIT License

  •7•138•5•0•Updated Apr 16, 2025Apr 16, 2025

• ActivePRM

  Public
  Jupyter Notebook

  •0•15•0•0•Updated Apr 16, 2025Apr 16, 2025

• understand-r1-zero

  Public
  Understanding R1-Zero-Like Training: A Critical Perspective

  rlreasoningllm+ 1r1-zero

  Python

  •

  MIT License

  •43•937•4•0•Updated Apr 15, 2025Apr 15, 2025

• dice

  Public
  Official implementation of Bootstrapping Language Models via DPO Implicit Rewards

  alignmentpreference-learninglarge-language-models+ 1rlhf

  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•238•0•0•Updated Apr 15, 2025Apr 15, 2025

• jax_xc

  Public
  Exchange correlation functionals translated from libxc to jax

  pythondensity-functional-theorylibxc+ 1jax

  Python

  •

  Mozilla Public License 2.0

  •2•45•4•0•Updated Mar 24, 2025Mar 24, 2025

• autofd

  Public
  Automatic Functional Differentiation in JAX

  automatic-differentiationjaxneural-operator+ 1variational-calculus

  Python

  •

  Apache License 2.0

  •1•72•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

  •7•121•0•0•Updated Mar 21, 2025Mar 21, 2025

• sailor2

  Public
  🔱 Sailor2: Sailing in South-East Asia with Inclusive Multilingual LLMs

  vietnameseindonesiathai+ 14tamiltagalogcebuanolanguage-modelburmesekhmerlao+ 7

  3•60•0•0•Updated Mar 21, 2025Mar 21, 2025

• Megatron-Sailor2

  Public
  Megatron for Sailor2/Qwen2.5

  Python

  •1•3•0•0•Updated Mar 21, 2025Mar 21, 2025

• SkyLadder

  Public
  The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling

  Python

  •

  Apache License 2.0

  •532•32•0•0•Updated Mar 20, 2025Mar 20, 2025

• AR-Video-Diffusion

  Public
  JavaScript

  •0•0•0•0•Updated Mar 17, 2025Mar 17, 2025

• Meta-ARVDM

  Public
  Official Implementation of "Error Analyses of Auto-Regressive Video Diffusion Models: A Unified Framework"

  0•5•1•0•Updated Mar 12, 2025Mar 12, 2025

• VocabularyParallelism

  Public
  Vocabulary Parallelism

  Python

  •

  Other

  •2.8k•19•2•0•Updated Mar 10, 2025Mar 10, 2025

• SEA-WildBench

  Public
  Multilingual WildBench for south-east Asian languages.

  Python

  •

  Apache License 2.0

  •0•2•1•0•Updated Mar 3, 2025Mar 3, 2025

• LongSpec

  Public
  LongSpec: Long-Context Speculative Decoding with Efficient Drafting and Verification

  Python

  •

  MIT License

  •2•53•0•0•Updated Mar 2, 2025Mar 2, 2025

• Rigging-ChatbotArena

  Public
  Improving Your Model Ranking on Chatbot Arena by Vote Rigging (ICML 2025)

  Python

  •2•20•0•0•Updated Feb 25, 2025Feb 25, 2025

• sailcraft

  Public
  🚢 Data Toolkit for Sailor Language Models

  data-deduplicationdata-cleaning

  Python

  •10•90•0•0•Updated Feb 24, 2025Feb 24, 2025

• EditAnything

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

  Python

  •

  Apache License 2.0

  •197•3.4k•45•0•Updated Feb 23, 2025Feb 23, 2025

• regmix

  Public
  [ICLR 2025] 🧬 RegMix: Data Mixture as Regression for Language Model Pre-training (Spotlight)

  Jupyter Notebook

  •

  MIT License

  •9•135•0•0•Updated Feb 17, 2025Feb 17, 2025

• sailcompass

  Public
  🧭 SailCompass: Towards Reproducible and Robust Evaluation for Southeast Asian Languages

  Python

  •1•13•0•0•Updated Feb 12, 2025Feb 12, 2025