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.
  ☆136Updated last year
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆188Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆281Updated 2 weeks ago
• gpu-mode / profiling-cuda-in-torch
  ☆162Updated last year
• eduardoleao052 / Autograd-from-scratch
  Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.
  ☆117Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆98Updated 6 months ago
• andylolu2 / simpleGEMM
  The simplest but fast implementation of matrix multiplication in CUDA.
  ☆37Updated 11 months ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆214Updated last year
• andreinechaev / nvcc4jupyter
  A plugin for Jupyter Notebook to run CUDA C/C++ code
  ☆237Updated 10 months ago
• lessw2020 / triton_kernels_for_fun_and_profit
  Custom kernels in Triton language for accelerating LLMs
  ☆23Updated last year
• google / aqt
  ☆322Updated 3 weeks ago
• SzymonOzog / FastSoftmax
  ☆47Updated 6 months ago
• HazyResearch / flash-fft-conv
  FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
  ☆323Updated 6 months ago
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆267Updated this week
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆101Updated last year
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆37Updated 2 years ago
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆151Updated last month
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆389Updated last week
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆188Updated last month
• google-research-datasets / tpu_graphs
  ☆125Updated last year
• tgautam03 / xGeMM
  Accelerated General (FP32) Matrix Multiplication from scratch in CUDA
  ☆120Updated 6 months ago
• salykova / sgemm.c
  Multi-Threaded FP32 Matrix Multiplication on x86 CPUs
  ☆350Updated 3 months ago
• CisMine / Guide-NVIDIA-Tools
  NVIDIA tools guide
  ☆140Updated 6 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆102Updated last year
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆291Updated 10 months ago
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆32Updated this week
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆47Updated this week
• VikParuchuri / triton_tutorial
  Tutorials for Triton, a language for writing gpu kernels
  ☆30Updated last year
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated 11 months ago