siboehm / ShallowSpeedLinks
Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
☆151Updated 2 years ago
Alternatives and similar repositories for ShallowSpeed
Users that are interested in ShallowSpeed are comparing it to the libraries listed below
Sorting:
- ☆256Updated this week
- Cataloging released Triton kernels.☆274Updated 2 months ago
- Fast low-bit matmul kernels in Triton☆401Updated last week
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆244Updated 6 months ago
- ☆177Updated last year
- ring-attention experiments☆160Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆196Updated 6 months ago
- Applied AI experiments and examples for PyTorch☆307Updated 3 months ago
- Learn CUDA with PyTorch☆117Updated last week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆61Updated this week
- A bunch of kernels that might make stuff slower 😉☆65Updated this week
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆195Updated 2 years ago
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆164Updated this week
- extensible collectives library in triton☆91Updated 8 months ago
- Ship correct and fast LLM kernels to PyTorch☆124Updated 2 weeks ago
- seqax = sequence modeling + JAX☆168Updated 4 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆584Updated 3 months ago
- Quantized LLM training in pure CUDA/C++.☆220Updated this week
- Learning about CUDA by writing PTX code.☆148Updated last year
- Fastest kernels written from scratch☆400Updated 2 months ago
- ☆337Updated last week
- Collection of kernels written in Triton language☆169Updated 7 months ago
- Custom kernels in Triton language for accelerating LLMs☆27Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆226Updated last year
- Experiment of using Tangent to autodiff triton☆80Updated last year
- Solve puzzles. Learn CUDA.☆64Updated last year
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆61Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆640Updated this week
- ☆546Updated last year
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆454Updated 3 weeks ago