Repositories list

• jrystal

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

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

  Python

  •0•3•4•0•Updated Mar 18, 2025Mar 18, 2025

• LightTrans

  Public
  The official implementation of "LightTransfer: Your Long-Context LLM is Secretly a Hybrid Model with Effortless Adaptation"

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

• AR-Video-Diffusion

  Public
  JavaScript

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

• TreeMeshGPT

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

  Python

  •

  MIT License

  •1•72•2•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•4•0•0•Updated Mar 12, 2025Mar 12, 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

  •13•222•3•0•Updated Mar 10, 2025Mar 10, 2025

• VocabularyParallelism

  Public
  Vocabulary Parallelism

  Python

  •

  Other

  •2.7k•17•0•0•Updated Mar 10, 2025Mar 10, 2025

• zero-bubble-pipeline-parallelism

  Public
  Zero Bubble Pipeline Parallelism

  Python

  •

  Other

  •2.7k•370•22•0•Updated Mar 4, 2025Mar 4, 2025

• LongSpec

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

  Python

  •

  MIT License

  •0•42•0•0•Updated Mar 2, 2025Mar 2, 2025

• Rigging-ChatbotArena

  Public
  Improving Your Model Ranking on Chatbot Arena by Vote Rigging

  Python

  •1•17•0•0•Updated Feb 25, 2025Feb 25, 2025

• d4ft

  Public
  A JAX library for Density Functional Theory.

  Python

  •

  Apache License 2.0

  •5•50•16•0•Updated Feb 25, 2025Feb 25, 2025

• sailcraft

  Public
  🚢 Data Toolkit for Sailor Language Models

  data-deduplicationdata-cleaning

  Python

  •10•87•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

  •196•3.4k•44•0•Updated Feb 23, 2025Feb 23, 2025

• sailor2

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

  vietnameseindonesiathai+ 14tamiltagalogcebuanolanguage-modelburmesekhmerlao+ 7

  3•52•0•0•Updated Feb 19, 2025Feb 19, 2025

• Megatron-Sailor2

  Public
  0•0•0•0•Updated Feb 19, 2025Feb 19, 2025

• regmix

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

  Jupyter Notebook

  •

  MIT License

  •6•115•0•0•Updated Feb 17, 2025Feb 17, 2025

• sailcompass

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

  Python

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

• oat-zero

  Public
  A lightweight reproduction of DeepSeek-R1-Zero with indepth analysis of self-reflection behavior.

  Python

  •

  MIT License

  •10•199•3•0•Updated Feb 6, 2025Feb 6, 2025

• autofd

  Public
  Automatic Functional Differentiation in JAX

  automatic-differentiationjaxneural-operator+ 1variational-calculus

  Python

  •

  Apache License 2.0

  •1•65•6•0•Updated Jan 17, 2025Jan 17, 2025

• I-FSJ

  Public
  Improved Few-Shot Jailbreaking Can Circumvent Aligned Language Models and Their Defenses (NeurIPS 2024)

  Jupyter Notebook

  •

  MIT License

  •9•58•1•0•Updated Jan 11, 2025Jan 11, 2025

• InfNeRF

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

  Python

  •

  Apache License 2.0

  •1•6•1•0•Updated Jan 7, 2025Jan 7, 2025

• sailor-llm

  Public
  [EMNLP-2024] ⚓️ Sailor: Open Language Models for South-East Asia

  indonesiathailanguage-model+ 4seavietnamlaomalay

  Python

  •

  MIT License

  •10•127•0•0•Updated Dec 21, 2024Dec 21, 2024

• Meta-Unlearning

  Public
  Python

  •1•26•2•0•Updated Dec 8, 2024Dec 8, 2024

• closer-look-LLM-unlearning

  Public
  The official code of the paper "A Closer Look at Machine Unlearning for Large Language Models".

  Python

  •5•23•1•0•Updated Dec 4, 2024Dec 4, 2024

• inceptionnext

  Public
  InceptionNeXt: When Inception Meets ConvNeXt (CVPR 2024)

  convolutional-neural-networks

  Python

  •

  Apache License 2.0

  •22•283•14•1•Updated Dec 2, 2024Dec 2, 2024

• stde

  Public
  Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024

  Python

  •6•102•0•0•Updated Nov 27, 2024Nov 27, 2024

• optim4rl

  Public
  Optim4RL is a Jax framework of learning to optimize for reinforcement learning.

  reinforcement-learningoptimizationoptimizer+ 9reinforcement-learning-algorithmsoptimization-algorithmsmeta-learningjaxlearning-to-learnoptimizersmeta-learning-algorithms+ 2

  Python

  •

  Apache License 2.0

  •2•24•0•0•Updated Nov 27, 2024Nov 27, 2024

• sdft

  Public
  [ACL 2024] The official codebase for the paper "Self-Distillation Bridges Distribution Gap in Language Model Fine-tuning".

  language-modelself-distillationsupervised-finetuning

  Shell

  •4•116•4•0•Updated Nov 2, 2024Nov 2, 2024

• Cheating-LLM-Benchmarks

  Public
  [ICLR 2025] Cheating Automatic LLM Benchmarks: Null Models Achieve High Win Rates (Oral)

  Jupyter Notebook

  •

  MIT License

  •0•73•0•0•Updated Oct 23, 2024Oct 23, 2024

• P-DoS

  Public
  [ArXiv 2024] Denial-of-Service Poisoning Attacks on Large Language Models

  Python

  •2•17•0•0•Updated Oct 22, 2024Oct 22, 2024