dlsyscourse / public_notebooks
☆58Updated 2 months ago
Alternatives and similar repositories for public_notebooks:
Users that are interested in public_notebooks are comparing it to the libraries listed below
- Examples and exercises from the book Programming Massively Parallel Processors - A Hands-on Approach. David B. Kirk and Wen-mei W. Hwu (T…☆64Updated 4 years ago
- ☆27Updated 9 months ago
- Learning about CUDA by writing PTX code.☆35Updated 11 months ago
- a minimal cache manager for PagedAttention, on top of llama3.☆67Updated 5 months ago
- the public repo for stats205 scribe notes at Stanford University☆13Updated 3 years ago
- ☆142Updated last year
- Cataloging released Triton kernels.☆168Updated last month
- Experiment of using Tangent to autodiff triton☆75Updated last year
- Learning material for CMU10-714: Deep Learning System☆233Updated 9 months ago
- ☆156Updated last year
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 11 months ago
- ring-attention experiments☆123Updated 4 months ago
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆67Updated 8 months ago
- ☆18Updated 4 years ago
- ☆128Updated last month
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆118Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 7 months ago
- ☆59Updated 3 months ago
- ☆22Updated last year
- ☆100Updated 5 months ago
- Collection of kernels written in Triton language☆105Updated this week
- Course materials for MIT6.5940: TinyML and Efficient Deep Learning Computing☆28Updated last month
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆89Updated this week
- ☆201Updated 2 months ago
- ☆214Updated last month
- ☆80Updated 4 months ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆84Updated last year
- A minimal implementation of vllm.☆33Updated 6 months ago
- Personal solutions to the Triton Puzzles☆17Updated 7 months ago
- Puzzles for learning Triton, play it with minimal environment configuration!☆229Updated 2 months ago