pytorch-labs / tritonparse
TritonParse is a new tool designed to help developers analyze and debug Triton kernels by visualizing the compilation process and source code mappings.
☆14Updated last week
Alternatives and similar repositories for tritonparse
Users that are interested in tritonparse are comparing it to the libraries listed below
Sorting:
- ☆13Updated 2 months ago
- Personal solutions to the Triton Puzzles☆18Updated 9 months ago
- Write a fast kernel and run it on Discord. See how you compare against the best!☆44Updated this week
- FlexAttention w/ FlashAttention3 Support☆26Updated 7 months ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated 2 months ago
- Make triton easier☆47Updated 11 months ago
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆22Updated last month
- Effective transpose on Hopper GPU☆18Updated 2 weeks ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆60Updated last month
- TORCH_LOGS parser for PT2☆37Updated 3 weeks ago
- extensible collectives library in triton☆86Updated last month
- A place to store reusable transformer components of my own creation or found on the interwebs☆55Updated this week
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆18Updated last week
- ☆18Updated last week
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆15Updated this week
- [WIP] Better (FP8) attention for Hopper☆30Updated 2 months ago
- ☆21Updated 2 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- High-Performance SGEMM on CUDA devices☆91Updated 3 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆124Updated this week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆43Updated this week
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated last month
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆85Updated this week
- Gpu benchmark☆61Updated 3 months ago
- PyTorch implementation of the Flash Spectral Transform Unit.☆16Updated 7 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated this week
- Exploration into the Firefly algorithm in Pytorch☆38Updated 3 months ago
- train with kittens!☆57Updated 6 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 9 months ago
- ☆14Updated last month