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.
  ☆118Updated last year
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆91Updated 9 months ago
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆169Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆252Updated 2 weeks ago
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆74Updated 3 weeks ago
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆170Updated this week
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆168Updated last month
• andylolu2 / simpleGEMM
  The simplest but fast implementation of matrix multiplication in CUDA.
  ☆34Updated 6 months ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆204Updated 11 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆142Updated last year
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆221Updated 6 months ago
• mobiusml / gemlite
  Fast low-bit matmul kernels in Triton
  ☆236Updated this week
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆161Updated 7 months ago
• CisMine / Guide-NVIDIA-Tools
  NVIDIA tools guide
  ☆102Updated last month
• google / aqt
  ☆284Updated last week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆303Updated this week
• salykova / matmul.c
  High-Performance FP32 Matrix Multiplication on CPU
  ☆333Updated this week
• vdesai2014 / inference-optimization-blog-post
  ☆86Updated 11 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆599Updated 7 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆107Updated 3 weeks ago
• rwitten / HighPerfLLMs2024
  ☆369Updated 7 months ago
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆681Updated last week
• CisMine / Parallel-Computing-Cuda-C
  CUDA Learning guide
  ☆326Updated 8 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆105Updated this week
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆18Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆179Updated last week
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆514Updated this week
• ulrichstern / cuda-convnet
  Alex Krizhevsky's original code from Google Code
  ☆189Updated 8 years ago
• lessw2020 / triton_kernels_for_fun_and_profit
  Custom kernels in Triton language for accelerating LLMs
  ☆17Updated 10 months ago
• 10-zin / cpp-micrograd
  A c/c++ implementation of micrograd: a tiny autograd engine with neural net on top.
  ☆63Updated last year