JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
☆67Feb 13, 2026Updated 2 weeks ago
Alternatives and similar repositories for jaxpp
Users that are interested in jaxpp are comparing it to the libraries listed below
Sorting:
- ☆13Jul 12, 2024Updated last year
- a size profiler for cuda binary☆72Jan 15, 2026Updated last month
- ☆37Jul 19, 2025Updated 7 months ago
- ☆19Mar 25, 2025Updated 11 months ago
- My study note for mlsys☆14Nov 4, 2024Updated last year
- A series of high-performance GEMM (General Matrix Multiply) implementations Iteratively optimised for H100 GPUs in Pure CUDA.☆70Feb 18, 2026Updated last week
- A home for the final text of all TVM RFCs.☆109Sep 24, 2024Updated last year
- The official implementation for Detector Guidance for Multi-Object Text-to-Image Generation (DG)☆20Feb 7, 2024Updated 2 years ago
- 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.☆18Apr 11, 2024Updated last year
- MLIR-based partitioning system☆167Updated this week
- Your favourite classical machine learning algos on the GPU/TPU☆22Dec 14, 2025Updated 2 months ago
- ☆25Jun 12, 2023Updated 2 years ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆234Sep 24, 2023Updated 2 years ago
- Open ABI and FFI for Machine Learning Systems☆351Updated this week
- ☆65Apr 26, 2025Updated 10 months ago
- Graph neural networks in JAX.☆68Jun 18, 2024Updated last year
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 7 months ago
- Autonomous GPU Kernel Generation & Optimization via Deep Agents☆248Updated this week
- A lightweight design for computation-communication overlap.☆223Jan 20, 2026Updated last month
- seqax = sequence modeling + JAX☆186Jul 23, 2025Updated 7 months ago
- NanoGPT (124M) quality in 2.67B tokens☆28Sep 17, 2025Updated 5 months ago
- Serving multiple LoRA finetuned LLM as one☆1,145May 8, 2024Updated last year
- Implementation of PSGD optimizer in JAX☆35Dec 31, 2024Updated last year
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆35Apr 25, 2024Updated last year
- Triton Compiler related materials.☆42Jan 4, 2025Updated last year
- Library for reading and processing ML training data.☆685Updated this week
- TL2cgen (TreeLite 2 C GENerator) is a model compiler for decision tree models☆46Feb 23, 2026Updated last week
- Port of the LLVM compiler infrastructure to the time-predictable processor Patmos☆15Apr 2, 2025Updated 11 months ago
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 2 months ago
- Valentine's Day Anonymous matching☆10Jul 25, 2014Updated 11 years ago
- The official starter-kit for NeurIPS 2025 mind games competition☆21Jul 27, 2025Updated 7 months ago
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- ☆74Sep 15, 2025Updated 5 months ago
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆446Feb 4, 2026Updated last month
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆46Dec 9, 2024Updated last year
- A Python Library for the 3GPP physical layer☆14Dec 18, 2025Updated 2 months ago
- ☆11Aug 23, 2023Updated 2 years ago
- Transparent serialization of python plain-old-data classes☆12Aug 31, 2022Updated 3 years ago
- PyTorch Quantization Framework For OCP MX Datatypes.☆16May 30, 2025Updated 9 months ago