Automatic differentiation for Triton Kernels
☆29Aug 12, 2025Updated last year
Alternatives and similar repositories for triton_bwd
Users that are interested in triton_bwd 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 Top-Down Profiler for GPU Applications☆24Feb 29, 2024Updated 2 years ago
- Triton kernels for Flux☆23Jul 7, 2025Updated last year
- This is a fork of SGLang for hip-attention integration. Please refer to hip-attention for detail.☆18Mar 31, 2026Updated 5 months ago
- Code for the ICML 2023 paper "Geometric Autoencoders - What You See is What You Decode"☆17Jul 28, 2023Updated 3 years ago
- Official Implementation of SEA: Sparse Linear Attention with Estimated Attention Mask (ICLR 2024)☆12Jun 20, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- A dynamic binary instrumentation tool for tracing and analyzing GPU kernel instructions.☆80Updated this week
- a size profiler for cuda binary☆71Jan 15, 2026Updated 8 months ago
- Fast low-bit matmul kernels in Triton☆486Aug 9, 2026Updated last month
- ☆100Sep 14, 2026Updated last week
- ☆19Nov 11, 2025Updated 10 months ago
- ☆356Updated this week
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆48Updated this week
- ☆14Apr 24, 2024Updated 2 years ago
- incubator repo for CUDA-TileIR backend☆162Updated 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.
- ☆40Dec 14, 2025Updated 9 months ago
- PyTorch implementation of the Flash Spectral Transform Unit.☆23Sep 19, 2024Updated 2 years ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆952Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆369Updated this week
- LES extension package for NequIP☆21Aug 14, 2026Updated last month
- A memory profiler for NVIDIA GPUs to explore memory inefficiencies in GPU-accelerated applications.☆38May 30, 2026Updated 3 months ago
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆203Updated this week
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 6 months ago
- Virtual Decoupled Cores: Composable Programming Framework and Runtime for Async GPUs☆22Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Supplementary material for our paper "Compute Trends Across Three Eras of Machine Learning".☆48Mar 12, 2022Updated 4 years ago
- Pytorch routines for (Ker)nel (Mac)hines☆11Oct 10, 2025Updated 11 months ago
- Cataloging released Triton kernels.☆312Sep 9, 2025Updated last year
- DeeperGEMM: crazy optimized version☆85May 5, 2025Updated last year
- pLiner is a framework that helps programmers identify locations in the source of numerical code that are highly affected by compiler opti…☆17Oct 27, 2023Updated 2 years ago
- Experiment of using Tangent to autodiff triton☆81Jan 22, 2024Updated 2 years ago
- ☆65Apr 26, 2025Updated last year
- Microbenchmarking hyperparameter tuning for JAX functions.☆24Updated this week
- FLA but cuTile☆27Apr 17, 2026Updated 5 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.
- Collection of kernels written in Triton language☆200Jan 27, 2026Updated 7 months ago
- A collection of GPU experiments and benchmarks for my personal understanding and research.☆43Aug 19, 2026Updated last month
- Triton for OpenCL backend, and use mlir-translate to get source OpenCL code☆27Aug 27, 2025Updated last year
- High-performance CUDA kernels for equivariant graph neural networks (MACE, NequIP, Allegro). 10-20x faster than e3nn.☆44Jan 17, 2026Updated 8 months ago
- A Triton-only attention backend for vLLM☆28Jul 14, 2026Updated 2 months ago
- An Optimizer for Nvidia Compilers.☆136Updated this week
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆22Nov 16, 2025Updated 10 months ago