pytorch-labs / helionLinks
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
☆166Updated this week
Alternatives and similar repositories for helion
Users that are interested in helion are comparing it to the libraries listed below
Sorting:
- extensible collectives library in triton☆86Updated 2 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆136Updated this week
- ☆63Updated this week
- ☆219Updated this week
- ☆81Updated 7 months ago
- ☆212Updated 11 months ago
- ☆91Updated 5 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆109Updated 11 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆90Updated 2 weeks ago
- ☆168Updated last year
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆48Updated last month
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆183Updated 4 months ago
- Fast low-bit matmul kernels in Triton☆322Updated this week
- Cataloging released Triton kernels.☆236Updated 5 months ago
- MLIR-based partitioning system☆91Updated this week
- Applied AI experiments and examples for PyTorch☆277Updated 3 weeks ago
- ☆115Updated last month
- Ahead of Time (AOT) Triton Math Library☆66Updated this week
- DeeperGEMM: crazy optimized version☆69Updated last month
- Collection of kernels written in Triton language☆128Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆99Updated 3 weeks ago
- Shared Middle-Layer for Triton Compilation☆255Updated this week
- A lightweight design for computation-communication overlap.☆141Updated last week
- ☆105Updated 9 months ago
- ☆109Updated 3 months ago
- Fastest kernels written from scratch☆281Updated 2 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆77Updated this week
- An experimental CPU backend for Triton☆126Updated 2 weeks ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 10 months ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆94Updated 6 years ago