andylolu2 / cuda-mnist

Related projects ⓘ

Alternatives and complementary repositories for cuda-mnist

• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆107Updated last year
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆86Updated 6 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆133Updated 9 months ago
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆145Updated this week
• lessw2020 / triton_kernels_for_fun_and_profit
  Custom kernels in Triton language for accelerating LLMs
  ☆17Updated 7 months ago
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆157Updated last year
• andylolu2 / simpleGEMM
  The simplest but fast implementation of matrix multiplication in CUDA.
  ☆33Updated 3 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆211Updated 3 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆153Updated this week
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆134Updated 2 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆215Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆68Updated 3 weeks ago
• vdesai2014 / inference-optimization-blog-post
  ☆82Updated 8 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆57Updated 11 months ago
• rasbt / pytorch-memory-optim
  This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…
  ☆86Updated last year
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆164Updated 3 months ago
• siboehm / SGEMM_CUDA
  Fast CUDA matrix multiplication from scratch
  ☆479Updated 10 months ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆483Updated 3 weeks ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆166Updated 3 weeks ago
• wentasah / mmul-anim
  Visualization of cache-optimized matrix multiplication
  ☆53Updated 5 years ago
• cchan / tccl
  extensible collectives library in triton
  ☆72Updated last month
• ulrichstern / cuda-convnet
  Alex Krizhevsky's original code from Google Code
  ☆190Updated 8 years ago
• thevasudevgupta / gpt-triton
  Triton implementation of GPT/LLAMA
  ☆16Updated 2 months ago
• apple / ml-recurrent-drafter
  ☆99Updated last month
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 5 months ago
• CisMine / Guide-NVIDIA-Tools
  NVIDIA tools guide
  ☆71Updated 3 months ago
• google / aqt
  ☆267Updated last week
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆14Updated 2 weeks ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆61Updated 11 months ago