JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
☆70Mar 11, 2026Updated 2 weeks ago
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☆71Jan 15, 2026Updated 2 months ago
- MLIR-based partitioning system☆174Updated this week
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 8 months ago
- A home for the final text of all TVM RFCs.☆109Sep 24, 2024Updated last year
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆79Dec 18, 2025Updated 3 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆51Mar 22, 2026Updated last week
- Minimal yet performant LLM examples in pure JAX☆246Jan 14, 2026Updated 2 months ago
- Open ABI and FFI for Machine Learning Systems☆365Updated this week
- A simple, performant and scalable JAX-based world modeling codebase.☆143Jan 15, 2026Updated 2 months ago
- ☆65Apr 26, 2025Updated 11 months ago
- ☆18Jan 3, 2025Updated last year
- Your favourite classical machine learning algos on the GPU/TPU☆22Dec 14, 2025Updated 3 months ago
- ICSE2021 Submission☆13Aug 28, 2022Updated 3 years ago
- MIPS simulator in C++☆13Oct 2, 2019Updated 6 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- ☆19Mar 25, 2025Updated last year
- Autonomous GPU Kernel Generation & Optimization via Deep Agents☆328Mar 18, 2026Updated last week
- A series of high-performance GEMM (General Matrix Multiply) implementations Iteratively optimised for H100 GPUs in Pure CUDA.☆73Feb 18, 2026Updated last month
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated 9 months ago
- seqax = sequence modeling + JAX☆187Jul 23, 2025Updated 8 months ago
- A project to apply a traditional implementation of Slurm on Kubernetes (with some magic)☆11Dec 20, 2017Updated 8 years ago
- TORCH_TRACE parser for PT2☆80Mar 17, 2026Updated last week
- Graph neural networks in JAX.☆67Jun 18, 2024Updated last year
- torchprime is a reference model implementation for PyTorch on TPU.☆46Mar 3, 2026Updated 3 weeks ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆18Mar 18, 2026Updated last week
- A lightweight design for computation-communication overlap.☆225Jan 20, 2026Updated 2 months ago
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 3 months ago
- ☆278Updated this week
- Serving multiple LoRA finetuned LLM as one☆1,148May 8, 2024Updated last year
- My study note for mlsys☆14Nov 4, 2024Updated last year
- ☆25Jun 12, 2023Updated 2 years ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated 6 months ago
- MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI tr…☆88Updated this week
- 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.
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆237Sep 24, 2023Updated 2 years ago
- ☆192Mar 28, 2023Updated 3 years ago
- Example of binding a TF32 CUTLASS GEMM kernel to PyTorch☆12Jun 7, 2024Updated last year
- A Python Library for the 3GPP physical layer☆15Dec 18, 2025Updated 3 months ago
- Simple intermediate representation language for learning and research.☆20Mar 27, 2020Updated 6 years ago
- A WebUI for Side-by-Side Comparison of Media (Images/Videos) Across Multiple Folders☆26Feb 21, 2025Updated last year
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month