A collection of megakernels for TPU
☆180Oct 15, 2026Updated 3 weeks ago
Alternatives and similar repositories for tpu-megakernels
Users that are interested in tpu-megakernels are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- FLAT: Flash Linear Attention on TPUs☆24Oct 13, 2026Updated 3 weeks ago
- ☆104Sep 8, 2026Updated last month
- ☆126Updated this week
- ☆102Updated this week
- mKernel: fast multi-node, multi-GPU fused kernels☆308Oct 17, 2026Updated 2 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.
- ☆157Updated this week
- ☆119Updated this week
- ☆15Jul 10, 2026Updated 3 months ago
- An agent harness that compiles a model into one provably-correct, self-retargeting CUDA megakernel and self-tunes it past cuBLAS at batch…☆161Sep 18, 2026Updated last month
- JAX backend for SGL☆374Updated this week
- my type shit☆21Aug 3, 2026Updated 3 months ago
- CODA: Rewriting Transformer Blocks as GEMM-Epilogue Programs☆258Updated this week
- A lightweight library for xPU kernel JIT compilation☆451Updated this week
- Tokamax: A GPU and TPU kernel library.☆325Updated this week
- 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.
- SGLang's TorchTPU Plugin☆74Updated this week
- ☆10Apr 14, 2026Updated 6 months ago
- Preview release of a unified serving engine, built for the fastest FastH3 serving.☆29Updated this week
- A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch☆386Jul 9, 2026Updated 3 months ago
- Curated high-performance TIRx GPU kernels☆35Updated this week
- DeepSelect: TopK kernels for DeepSeek Sparse Attention (DSA) and Samplers☆533Sep 30, 2026Updated last month
- Moving to vllm-project/vllm-jax. TPU inference for vLLM, with unified JAX and PyTorch support.☆477Updated this week
- MoonEP: A Perfectly Balanced Expert Parallelism Library via Dynamic Redundant Experts☆1,202Oct 12, 2026Updated 3 weeks ago
- Automatic formal verification for warp-specialized GPU kernels.☆109Updated this week
- 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.
- High-performance Rust benchmark client for vLLM serving endpoints.☆52Aug 3, 2026Updated 3 months ago
- ☆138Updated this week
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆106Apr 7, 2026Updated 6 months ago
- Reference implementation and examples of the CuTe Layout representation and algebra.☆392Updated this week
- Kernel Design Agents (KDA) is a agent-centric workflow to write high-performance CUDA Kernels.☆1,541Oct 19, 2026Updated 2 weeks ago
- ☆37Aug 7, 2025Updated last year
- Dist-MoE implements Speed-of-Light distributed Mixture-of-Experts forward and backward kernels. They are highly optimized and fused comm-…☆69Oct 29, 2026Updated last week
- Distributed MoE in a Single Kernel [NeurIPS '25]☆332May 5, 2026Updated 6 months ago
- Official repository for Parallax (Parameterized Local Linear Attention)☆71Jul 30, 2026Updated 3 months ago
- 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.
- Autonomous CUDA kernel optimization agent with structured task specs and per-config scoring☆19Jun 17, 2026Updated 4 months ago
- ☆27Sep 10, 2026Updated last month
- TPUMS (TPU Microbenchmark Suite) is a modular, high-fidelity benchmarking and profiling framework designed to evaluate the compute, memor…☆38Updated this week
- NaiveRT: a self-contained TP8 inference runtime for the Naive N series on eight Blackwell GPUs with UMMA support. One persistent CUDA gra…☆135Sep 27, 2026Updated last month
- ☆141Updated this week
- ☆14Jul 13, 2025Updated last year
- A PyTorch native automated mega-kernel compiler☆32Updated this week