Alternatives and similar repositories for axonn

Users that are interested in axonn are comparing it to the libraries listed below

• cchan / tccl
  extensible collectives library in triton
  ☆86Updated 2 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• Jokeren / triton-samples
  ☆28Updated 5 months ago
• triton-lang / kernels
  ☆81Updated 7 months ago
• Dao-AILab / quack
  A Quirky Assortment of CuTe Kernels
  ☆117Updated this week
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆99Updated last year
• stanford-futuredata / stk
  ☆105Updated 10 months ago
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆46Updated this week
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆132Updated 2 months ago
• open-lm-engine / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆51Updated this week
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆27Updated this week
• spcl / sten
  Sparsity support for PyTorch
  ☆35Updated 3 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated 10 months ago
• MekkCyber / CutlassAcademy
  A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS
  ☆189Updated last month
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆95Updated 5 months ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆134Updated last year
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆19Updated 11 months ago
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆45Updated 11 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆100Updated 3 weeks ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 3 months ago
• simveit / effective_transpose
  Effective transpose on Hopper GPU
  ☆23Updated last month
• pytorch-labs / monarch
  PyTorch Single Controller
  ☆231Updated this week
• osayamenja / Kleos
  Complete GPU residency for ML.
  ☆17Updated last week
• mgmalek / efficient_cross_entropy
  ☆109Updated last year
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆65Updated 3 years ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆167Updated this week
• gpu-mode / ring-attention
  ring-attention experiments
  ☆144Updated 8 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆219Updated this week
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆166Updated this week
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆23Updated 4 years ago