JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
☆77Jun 3, 2026Updated this week
Alternatives and similar repositories for jaxpp
Users that are interested in jaxpp are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- a size profiler for cuda binary☆70Jan 15, 2026Updated 4 months ago
- ☆13Jul 12, 2024Updated last year
- MLIR-based partitioning system☆186May 31, 2026Updated last week
- ☆38Jul 19, 2025Updated 10 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated 10 months 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.
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- A home for the final text of all TVM RFCs.☆110Sep 24, 2024Updated last year
- Minimal yet performant LLM examples in pure JAX☆256Apr 10, 2026Updated last month
- Open ABI and FFI for Machine Learning Systems☆404Updated this week
- A simple, performant and scalable JAX-based world modeling codebase.☆151Jan 15, 2026Updated 4 months ago
- ☆66Apr 26, 2025Updated last year
- The AI-First Software Development (AIFSD) Manifesto☆29Feb 17, 2026Updated 3 months ago
- 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.☆18Apr 11, 2024Updated 2 years ago
- [NeurIPS'25] KVCOMM: Online Cross-context KV-cache Communication for Efficient LLM-based Multi-agent Systems☆16Nov 1, 2025Updated 7 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆19Mar 25, 2025Updated last year
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- seqax = sequence modeling + JAX☆191Jul 23, 2025Updated 10 months ago
- TORCH_TRACE parser for PT2☆86May 11, 2026Updated 3 weeks ago
- See what your HTML looks like☆46Oct 20, 2020Updated 5 years ago
- Graph neural networks in JAX.☆67Jun 18, 2024Updated last year
- torchprime is a reference model implementation for PyTorch on TPU.☆47Mar 3, 2026Updated 3 months ago
- Microsoft Linear Algebra Subroutines☆14Oct 18, 2025Updated 7 months ago
- Autonomous GPU Kernel Generation & Optimization via Deep Agents☆436May 28, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆18Apr 17, 2026Updated last month
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- A lightweight design for computation-communication overlap.☆234Jan 20, 2026Updated 4 months ago
- Serving multiple LoRA finetuned LLM as one☆1,160May 8, 2024Updated 2 years ago
- ☆302Updated this week
- TL2cgen (TreeLite 2 C GENerator) is a model compiler for decision tree models☆48May 25, 2026Updated last week
- This is the official Python version of CoreInfer: Accelerating Large Language Model Inference with Semantics-Inspired Adaptive Sparse Act…☆17Oct 25, 2024Updated last year
- ☆25Jun 12, 2023Updated 2 years ago
- NanoGPT (124M) quality in 2.67B tokens☆28Sep 17, 2025Updated 8 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- [CVPR'24] Once for Both: Single Stage of Importance and Sparsity Search for Vision Transformer Compression☆15Jul 1, 2024Updated last year
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated 8 months ago
- Ongoing research training transformer models at scale☆18Updated this week
- PyTorch Quantization Framework For OCP MX Datatypes.☆16May 30, 2025Updated last year
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆246Sep 24, 2023Updated 2 years ago
- Library for reading and processing ML training data.☆740Updated this week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆487May 19, 2026Updated 2 weeks ago