srush / GPU-Puzzles
Solve puzzles. Learn CUDA.
☆5,703Updated 2 weeks ago
Related projects: ⓘ
- Solve puzzles. Improve your pytorch.☆3,056Updated 2 months ago
- The full minitorch student suite.☆1,849Updated last month
- A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API☆10,011Updated last month
- Machine Learning Engineering Open Book☆10,986Updated this week
- Explanation to key concepts in ML☆7,029Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,362Updated this week
- Supercharge Your Model Training☆5,116Updated this week
- Material for cuda-mode lectures☆2,401Updated 2 weeks ago
- CUDA related news and material links☆1,079Updated 2 weeks ago
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,213Updated 10 months ago
- Development repository for the Triton language and compiler☆12,698Updated this week
- Puzzles for learning Triton☆966Updated this week
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆4,573Updated last week
- It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; ho…☆4,504Updated 2 years ago
- 🔥Highlighting the top ML papers every week.☆9,910Updated last week
- Neural Networks: Zero to Hero☆11,524Updated last month
- Fast and memory-efficient exact attention☆13,401Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.☆5,950Updated this week
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆5,145Updated this week
- An autoregressive character-level language model for making more things☆2,471Updated 3 months ago
- arxiv-sanity lite: tag arxiv papers of interest get recommendations of similar papers in a nice UI using SVMs over tfidf feature vectors …☆1,156Updated last year
- ManimML is a project focused on providing animations and visualizations of common machine learning concepts with the Manim Community Libr…☆2,333Updated 2 months ago
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆19,845Updated last month
- The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.☆9,410Updated last week
- Pen and paper exercises in machine learning☆1,929Updated 3 months ago
- Ongoing research training transformer models at scale☆9,949Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,577Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆7,687Updated this week
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,831Updated 3 months ago
- CUDA Templates for Linear Algebra Subroutines☆5,359Updated this week