☆136Dec 9, 2025Updated 3 months ago
Alternatives and similar repositories for lmpo
Users that are interested in lmpo are comparing it to the libraries listed below
Sorting:
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 9 months ago
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Oct 6, 2021Updated 4 years ago
- Code for Scalable Offline Model-Based RL with Action chunking☆19Feb 20, 2026Updated 2 weeks ago
- Code for "What really matters in matrix-whitening optimizers?"☆22Oct 31, 2025Updated 4 months ago
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 2 years ago
- Answers to the questions at the back of the chapters of Advances in Financial Machine Learning.☆23Apr 11, 2020Updated 5 years ago
- ☆13Jul 2, 2025Updated 8 months ago
- Code for Tackling Long-Horizon Tasks with Model-based Offline Reinforcement Learning☆16Feb 6, 2025Updated last year
- Offline RL algoritms implemented in Stable Baselines3 (pytorch)☆10Dec 7, 2021Updated 4 years ago
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 4 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Jul 24, 2025Updated 7 months ago
- ☆22May 5, 2025Updated 10 months ago
- The official implementation of "Horizon Reduction Makes RL Scalable"☆181Aug 2, 2025Updated 7 months ago
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆599Oct 7, 2025Updated 5 months ago
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆142Jun 23, 2025Updated 8 months ago
- Crawl & visualize ICLR papers and reviews.☆18Nov 5, 2022Updated 3 years ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated last year
- Code for Evaluating Explanations for Reading Comprehension with Realistic Counterfactuals.☆18Apr 25, 2021Updated 4 years ago
- Digital Trigger Unit for V2&V3 Gearbox☆15Jan 11, 2021Updated 5 years ago
- ☆16Oct 5, 2021Updated 4 years ago
- [AAMAS'26] xTED: Cross-Domain Adaptation via Diffusion-Based Trajectory Editing☆24Jan 8, 2026Updated 2 months ago
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆21Oct 6, 2021Updated 4 years ago
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- Explaining neural decisions contrastively to alternative decisions.☆25Mar 18, 2021Updated 4 years ago
- Code for "Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretraining"☆27Oct 14, 2025Updated 4 months ago
- ☆92May 31, 2025Updated 9 months ago
- COOM: Benchmarking Continual Reinforcement Learning on Doom☆21Jan 4, 2026Updated 2 months ago
- Drop-in environment replacements that make your RL algorithm train faster.☆21Jun 19, 2024Updated last year
- AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)☆33Sep 30, 2025Updated 5 months ago
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- This is the official implementation for the paper "Learning to Scaffold: Optimizing Model Explanations for Teaching"☆20May 19, 2022Updated 3 years ago
- Scalable Opponent Shaping Experiments in JAX☆25Apr 13, 2024Updated last year
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆24Sep 29, 2024Updated last year
- ☆45Feb 27, 2026Updated last week
- Code for paper "Leakage-Adjusted Simulatability: Can Models Generate Non-Trivial Explanations of Their Behavior in Natural Language?"☆22Oct 13, 2020Updated 5 years ago
- Storing long contexts in tiny caches with self-study☆244Dec 5, 2025Updated 3 months ago
- Definitive implementation of the stochastic interpolant framework for generative modeling in jax.☆40Jul 13, 2025Updated 7 months ago
- Foundation Policies with Hilbert Representations (ICML 2024)☆105Sep 29, 2025Updated 5 months ago