SzymonOzog / FastSoftmax
☆25Updated last month
Alternatives and similar repositories for FastSoftmax:
Users that are interested in FastSoftmax are comparing it to the libraries listed below
- ☆27Updated last week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆17Updated this week
- Cataloging released Triton kernels.☆167Updated last month
- Fast low-bit matmul kernels in Triton☆232Updated this week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆117Updated last year
- Collection of kernels written in Triton language☆103Updated this week
- Triton implementation of GPT/LLAMA☆16Updated 5 months ago
- High-Performance SGEMM on CUDA devices☆74Updated 3 weeks ago
- ☆86Updated 11 months ago
- Fastest kernels written from scratch☆162Updated this week
- ☆141Updated last year
- extensible collectives library in triton☆82Updated 4 months ago
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆172Updated last week
- Experiment of using Tangent to autodiff triton☆75Updated last year
- The simplest but fast implementation of matrix multiplication in CUDA.☆34Updated 6 months ago
- Make triton easier☆44Updated 8 months ago
- JAX bindings for Flash Attention v2☆85Updated 7 months ago
- ☆177Updated this week
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆37Updated this week
- Learn CUDA with PyTorch☆16Updated 2 weeks ago
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆229Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆221Updated 6 months ago
- Accelerated First Order Parallel Associative Scan☆171Updated 5 months ago
- ring-attention experiments☆123Updated 4 months ago
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Updated 6 months ago
- supporting pytorch FSDP for optimizers☆76Updated 2 months ago
- Applied AI experiments and examples for PyTorch☆224Updated this week
- ☆75Updated 7 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 5 months ago