Jokeren / triton-samplesLinks
☆28Updated 6 months ago
Alternatives and similar repositories for triton-samples
Users that are interested in triton-samples are comparing it to the libraries listed below
Sorting:
- extensible collectives library in triton☆87Updated 3 months ago
- Experiment of using Tangent to autodiff triton☆79Updated last year
- A bunch of kernels that might make stuff slower 😉☆55Updated this week
- ☆21Updated 4 months ago
- Personal solutions to the Triton Puzzles☆19Updated last year
- Write a fast kernel and run it on Discord. See how you compare against the best!☆46Updated this week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆112Updated last year
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆187Updated this week
- ☆106Updated 10 months ago
- ☆169Updated last year
- Sparsity support for PyTorch☆35Updated 3 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆91Updated 2 weeks ago
- Collection of kernels written in Triton language☆136Updated 3 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆79Updated last week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆51Updated last week
- TORCH_LOGS parser for PT2☆46Updated this week
- ☆225Updated last week
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Ahead of Time (AOT) Triton Math Library☆70Updated last week
- ☆39Updated last year
- ☆74Updated 3 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆135Updated last year
- Explore training for quantized models☆20Updated this week
- MLIR-based partitioning system☆103Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆80Updated 10 months ago
- ☆26Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆158Updated 3 weeks ago
- a minimal cache manager for PagedAttention, on top of llama3.☆93Updated 10 months ago
- Make triton easier☆47Updated last year
- ring-attention experiments☆143Updated 9 months ago