JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
☆78Jun 18, 2026Updated last 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☆69Jan 15, 2026Updated 5 months ago
- MLIR-based partitioning system☆191Updated this week
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated 10 months ago
- 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
- 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.
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆84Dec 18, 2025Updated 6 months ago
- Minimal yet performant LLM examples in pure JAX☆261Apr 10, 2026Updated 2 months ago
- Open ABI and FFI for Machine Learning Systems☆418Updated this week
- A simple, performant and scalable JAX-based world modeling codebase.☆151Jan 15, 2026Updated 5 months ago
- ☆65Apr 26, 2025Updated last year
- ICSE2021 Submission☆13Aug 28, 2022Updated 3 years ago
- Decorators for logging purposes for all your dataframes☆15Jan 31, 2025Updated last year
- ☆19Mar 25, 2025Updated last year
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated last year
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- For building the world's largest dataset of GPU kernels.☆10Jun 11, 2026Updated 2 weeks ago
- seqax = sequence modeling + JAX☆194Jul 23, 2025Updated 11 months ago
- Official repository for "TrustGeoGen: Formal-Verified Data Engine for Trustworthy Multi-modal Geometric Problem Solving"☆23Sep 1, 2025Updated 9 months ago
- ☆13Apr 11, 2025Updated last year
- A project to apply a traditional implementation of Slurm on Kubernetes (with some magic)☆11Dec 20, 2017Updated 8 years ago
- A series of high-performance GEMM (General Matrix Multiply) implementations Iteratively optimised for H100 GPUs in Pure CUDA.☆79Feb 18, 2026Updated 4 months ago
- TORCH_TRACE parser for PT2☆87May 11, 2026Updated last month
- Graph neural networks in JAX.☆67Jun 18, 2024Updated 2 years ago
- Autonomous GPU Kernel Generation & Optimization via Deep Agents☆455Jun 6, 2026Updated 3 weeks ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- Berkeley OS Prelim Reading Notes☆15Sep 20, 2023Updated 2 years ago
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 6 months ago
- A lightweight design for computation-communication overlap.☆241Jan 20, 2026Updated 5 months ago
- Serving multiple LoRA finetuned LLM as one☆1,162May 8, 2024Updated 2 years ago
- My study note for mlsys☆14Nov 4, 2024Updated last year
- ☆25Jun 12, 2023Updated 3 years ago
- NanoGPT (124M) quality in 2.67B tokens☆28Sep 17, 2025Updated 9 months ago
- [CVPR'24] Once for Both: Single Stage of Importance and Sparsity Search for Vision Transformer Compression☆15Jul 1, 2024Updated last year
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated 9 months ago
- 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
- ☆192Mar 28, 2023Updated 3 years ago
- Library for reading and processing ML training data.☆747Updated this week
- Simple intermediate representation language for learning and research.☆22Mar 27, 2020Updated 6 years ago
- Stainless neural networks in JAX☆36Jun 13, 2026Updated 2 weeks ago