IaroslavElistratov / triton-autodiffLinks
☆18Updated last month
Alternatives and similar repositories for triton-autodiff
Users that are interested in triton-autodiff are comparing it to the libraries listed below
Sorting:
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆67Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆64Updated 2 weeks ago
- ☆12Updated 3 months ago
- Ship correct and fast LLM kernels to PyTorch☆126Updated this week
- PCCL (Prime Collective Communications Library) implements fault tolerant collective communications over IP☆141Updated 3 months ago
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆172Updated last week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆153Updated 2 years ago
- Automatic differentiation for Triton Kernels☆29Updated 4 months ago
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆196Updated 2 years ago
- High-Performance SGEMM on CUDA devices☆113Updated 10 months ago
- Quantized LLM training in pure CUDA/C++.☆221Updated this week
- ☆28Updated 10 months ago
- ☆14Updated 9 months ago
- Learning about CUDA by writing PTX code.☆149Updated last year
- train with kittens!☆63Updated last year
- extensible collectives library in triton☆91Updated 8 months ago
- TORCH_LOGS parser for PT2☆69Updated last month
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆178Updated this week
- ☆177Updated last year
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆66Updated 8 months ago
- ☆262Updated this week
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆84Updated 3 weeks ago
- ☆90Updated last year
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆58Updated 3 weeks ago
- A bunch of kernels that might make stuff slower 😉☆65Updated last week
- Experiment of using Tangent to autodiff triton☆81Updated last year
- SIMD quantization kernels☆93Updated 3 months ago
- Collection of kernels written in Triton language☆173Updated 8 months ago
- Fast and Furious AMD Kernels☆314Updated 2 weeks ago
- Triton-based Symmetric Memory operators and examples☆65Updated last month