Alternatives and similar repositories for cuda-mnist

Users that are interested in cuda-mnist are comparing it to the libraries listed below

• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆138Updated last year
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆189Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆293Updated 3 weeks ago
• gpu-mode / profiling-cuda-in-torch
  ☆163Updated last year
• eduardoleao052 / Autograd-from-scratch
  Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.
  ☆120Updated last year
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆391Updated last week
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆72Updated last week
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆154Updated 2 months ago
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆192Updated 3 months ago
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆279Updated last month
• srush / Autodiff-Puzzles
  ☆450Updated 10 months ago
• andylolu2 / simpleGEMM
  The simplest but fast implementation of matrix multiplication in CUDA.
  ☆38Updated last year
• lessw2020 / triton_kernels_for_fun_and_profit
  Custom kernels in Triton language for accelerating LLMs
  ☆25Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆216Updated last year
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆63Updated last year
• andreinechaev / nvcc4jupyter
  A plugin for Jupyter Notebook to run CUDA C/C++ code
  ☆240Updated 11 months ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆291Updated last year
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆54Updated this week
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆145Updated 2 weeks ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆63Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆99Updated 7 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆718Updated last month
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆577Updated this week
• modular / mojo-gpu-puzzles
  Learn GPU Programming in Mojo🔥 by Solving Puzzles
  ☆124Updated 2 weeks ago
• huggingface / optimum-graphcore
  Blazing fast training of 🤗 Transformers on Graphcore IPUs
  ☆86Updated last year
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆140Updated 4 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆288Updated last month
• leokruglikov / CUDA-notes
  Personal notes on CUDA programming
  ☆55Updated 2 years ago