☆142Dec 9, 2025Updated 9 months ago
Alternatives and similar repositories for lmpo
Users that are interested in lmpo are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code for "What really matters in matrix-whitening optimizers?"☆25Oct 31, 2025Updated 11 months ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆16May 28, 2025Updated last year
- Minimal but scalable implementation of large language models in JAX☆34Nov 28, 2025Updated 10 months ago
- Ludic – an LLM-RL library for the era of experience☆67Aug 9, 2026Updated 2 months ago
- Minimal yet performant LLM examples in pure JAX☆287Aug 23, 2026Updated last month
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A benchmark for offline goal-conditioned RL and offline RL☆498Updated this week
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆97Dec 1, 2024Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated last year
- Implementation of PSGD optimizer in JAX☆41Dec 31, 2024Updated last year
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆39May 19, 2023Updated 3 years ago
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated 2 years ago
- Offline RL algoritms implemented in Stable Baselines3 (pytorch)☆11Dec 7, 2021Updated 4 years ago
- Storing long contexts in tiny caches with self-study☆340Mar 23, 2026Updated 6 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Code for the paper Normalizing Flows are Capable Models for RL☆20Jun 3, 2025Updated last year
- DINO-based perceptual losses and FDD feature extraction☆39Jan 7, 2026Updated 9 months ago
- ☆12Apr 25, 2022Updated 4 years ago
- Dreamer 4 jax implementation☆106Jul 24, 2026Updated 2 months ago
- A simple, performant and scalable JAX-based world modeling codebase.☆164Jan 15, 2026Updated 8 months ago
- Code to accompany the paper "The Information Geometry of Unsupervised Reinforcement Learning"☆20Oct 6, 2021Updated 5 years ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated last year
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆288Sep 21, 2026Updated 2 weeks ago
- Kinetics: Rethinking Test-Time Scaling Laws☆87Jul 11, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆635Oct 7, 2025Updated last year
- Self-Questioning Language Models☆56Mar 30, 2026Updated 6 months ago
- ☆127Jun 11, 2025Updated last year
- Scale digital agent rollouts without pain.☆36Jun 18, 2026Updated 3 months ago
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆158Apr 7, 2026Updated 6 months ago
- Vortex: Programmable Sparse Attention for Agents as Algorithm Designers☆69Aug 22, 2026Updated last month
- Code for Tackling Long-Horizon Tasks with Model-based Offline Reinforcement Learning☆19Feb 6, 2025Updated last year
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 years ago
- Open-source framework for the research and development of foundation models.☆3,893Updated this week
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆29May 11, 2026Updated 4 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆29Jan 14, 2025Updated last year
- Code for Scalable Offline Model-Based RL with Action chunking☆34Feb 20, 2026Updated 7 months ago
- Decoupled Q-Chunking☆75May 3, 2026Updated 5 months ago
- Code and data for paper "(How) do Language Models Track State?"☆29Mar 31, 2025Updated last year
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆128Jan 10, 2026Updated 8 months ago
- SkyRL: A Modular Full-stack RL Library for LLMs☆2,386Updated this week