tgautam03 / tGeMM
General Matrix Multiplication using NVIDIA Tensor Cores
☆13Updated 3 months ago
Alternatives and similar repositories for tGeMM:
Users that are interested in tGeMM are comparing it to the libraries listed below
- Write a fast kernel and run it on Discord. See how you compare against the best!☆41Updated this week
- High-Performance SGEMM on CUDA devices☆90Updated 3 months ago
- Custom PTX Instruction Benchmark☆122Updated last month
- LLM training in simple, raw C/CUDA☆92Updated 11 months ago
- Learning about CUDA by writing PTX code.☆128Updated last year
- Custom kernels in Triton language for accelerating LLMs☆18Updated last year
- Experiment of using Tangent to autodiff triton☆78Updated last year
- ☆13Updated last month
- Learn CUDA with PyTorch☆20Updated 2 months ago
- ☆27Updated 9 months ago
- making the official triton tutorials actually comprehensible☆26Updated last month
- Attention in SRAM on Tenstorrent Grayskull☆34Updated 9 months ago
- Reference Kernels for the Leaderboard☆33Updated last week
- Experimental GPU language with meta-programming☆22Updated 7 months ago
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆65Updated last month
- Jax like function transformation engine but micro, microjax☆30Updated 6 months ago
- Custom triton kernels for training Karpathy's nanoGPT.☆18Updated 6 months ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated 10 months ago
- A parallel framework for training deep neural networks☆58Updated last month
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 9 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆62Updated this week
- Personal solutions to the Triton Puzzles☆18Updated 9 months ago
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Updated 8 months ago
- Explore training for quantized models☆17Updated 3 months ago
- ☆102Updated this week
- ring-attention experiments☆130Updated 6 months ago
- ML/DL Math and Method notes☆60Updated last year
- ☆31Updated 3 months ago
- This repository contain the simple llama3 implementation in pure jax.☆63Updated 2 months ago
- Accelerated General (FP32) Matrix Multiplication from scratch in CUDA☆114Updated 3 months ago