siboehm / ShallowSpeed
Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
☆125Updated last year
Alternatives and similar repositories for ShallowSpeed:
Users that are interested in ShallowSpeed are comparing it to the libraries listed below
- Cataloging released Triton kernels.☆185Updated 2 months ago
- Applied AI experiments and examples for PyTorch☆243Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆31Updated this week
- Collection of kernels written in Triton language☆110Updated 3 weeks ago
- ☆188Updated 3 weeks ago
- Fast low-bit matmul kernels in Triton☆257Updated this week
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆227Updated this week
- Fastest kernels written from scratch☆188Updated this week
- extensible collectives library in triton☆83Updated 5 months ago
- Experiment of using Tangent to autodiff triton☆76Updated last year
- ☆148Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- ring-attention experiments☆127Updated 4 months ago
- seqax = sequence modeling + JAX☆148Updated 2 weeks ago
- ☆73Updated 4 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆520Updated 3 weeks ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆188Updated this week
- ☆100Updated 6 months ago
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆173Updated last year
- ☆86Updated last year
- The simplest but fast implementation of matrix multiplication in CUDA.☆35Updated 7 months ago
- Solve puzzles. Learn CUDA.☆63Updated last year
- ☆161Updated 8 months ago
- A library for unit scaling in PyTorch☆124Updated 3 months ago
- ☆212Updated 7 months ago
- ☆75Updated 8 months ago
- High-Performance SGEMM on CUDA devices☆86Updated last month
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆231Updated 2 weeks ago