pytorch / helionLinks
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
☆328Updated 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☆88Updated 6 months ago
- ☆242Updated this week
- A Quirky Assortment of CuTe Kernels☆603Updated last week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆230Updated last week
- ☆177Updated last year
- Applied AI experiments and examples for PyTorch☆296Updated last month
- Cataloging released Triton kernels.☆261Updated last month
- ☆90Updated 11 months ago
- Fast low-bit matmul kernels in Triton☆376Updated last week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆54Updated this week
- ☆121Updated 9 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- ☆238Updated last year
- A library to analyze PyTorch traces.☆415Updated this week
- TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer for Triton Kernels☆152Updated this week
- Fastest kernels written from scratch☆366Updated 3 weeks ago
- Collection of kernels written in Triton language☆157Updated 6 months ago
- Github mirror of trition-lang/triton repo.☆82Updated this week
- ☆331Updated 3 weeks ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆213Updated last week
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆328Updated 3 weeks ago
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆82Updated this week
- kernels, of the mega variety☆563Updated last week
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆230Updated 5 months ago
- Shared Middle-Layer for Triton Compilation☆289Updated last week
- PyTorch bindings for CUTLASS grouped GEMM.☆123Updated 4 months ago
- A bunch of kernels that might make stuff slower 😉☆59Updated this week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆373Updated this week
- ☆113Updated last year
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆592Updated this week