☆143Dec 9, 2025Updated 10 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:
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆16May 28, 2025Updated last year
- This repository, deepspeed-grpo-qlora-vllm, provides a complete framework for fine-tuning LLMs using Group Relative Policy Optimization (…☆13Jul 15, 2025Updated last year
- The official implementation of "Horizon Reduction Makes RL Scalable"☆208Aug 2, 2025Updated last year
- Minimal but scalable implementation of large language models in JAX☆34Nov 28, 2025Updated 11 months ago
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Minimal yet performant LLM examples in pure JAX☆286Aug 23, 2026Updated 2 months ago
- A benchmark for offline goal-conditioned RL and offline RL☆507Updated 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 https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated 2 years ago
- ☆13Apr 25, 2022Updated 4 years ago
- Code for the paper Normalizing Flows are Capable Models for RL☆20Jun 3, 2025Updated last year
- Implementation of PSGD optimizer in JAX☆41Dec 31, 2024Updated last year
- Offline RL algoritms implemented in Stable Baselines3 (pytorch)☆11Dec 7, 2021Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Storing long contexts in tiny caches with self-study☆344Mar 23, 2026Updated 7 months ago
- Dreamer 4 jax implementation☆106Jul 24, 2026Updated 3 months ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Oct 21, 2024Updated 2 years ago
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆640Oct 7, 2025Updated last year
- Kinetics: Rethinking Test-Time Scaling Laws☆87Jul 11, 2025Updated last year
- Self-Questioning Language Models☆55Mar 30, 2026Updated 6 months ago
- ☆127Jun 11, 2025Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆165Jan 15, 2026Updated 9 months ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆40May 19, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 years ago
- ☆21Apr 22, 2026Updated 6 months ago
- Code for Tackling Long-Horizon Tasks with Model-based Offline Reinforcement Learning☆19Feb 6, 2025Updated last year
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆160Apr 7, 2026Updated 6 months ago
- ☆22May 27, 2024Updated 2 years ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆29Jan 14, 2025Updated last year
- Q-learning with Adjoint Matching☆131May 11, 2026Updated 5 months ago
- NanoGPT (124M) in 90 seconds☆5,976Sep 28, 2026Updated last month
- Decoupled Q-Chunking☆75May 3, 2026Updated 5 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Vortex: Programmable Sparse Attention for Agents as Algorithm Designers☆69Aug 22, 2026Updated 2 months ago
- ☆29May 11, 2026Updated 5 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆718Jan 26, 2026Updated 9 months ago
- Scale digital agent rollouts without pain.☆38Jun 18, 2026Updated 4 months ago
- seqax = sequence modeling + JAX☆199Jul 23, 2025Updated last year
- ☆91Aug 4, 2025Updated last year
- ☆34Jun 21, 2024Updated 2 years ago