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 last month
• Jokeren / triton-samples
  ☆27Updated 3 months ago
• spcl / sten
  Sparsity support for PyTorch
  ☆34Updated last month
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• ScalingIntelligence / hydragen
  Hydragen: High-Throughput LLM Inference with Shared Prefixes
  ☆36Updated last year
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆94Updated last year
• triton-lang / kernels
  ☆79Updated 6 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated 9 months ago
• simveit / effective_transpose
  Effective transpose on Hopper GPU
  ☆17Updated last week
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆90Updated 3 months ago
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆44Updated last week
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆88Updated last week
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆122Updated this week
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆115Updated 5 months ago
• stanford-futuredata / stk
  ☆104Updated 8 months ago
• deepspeedai / DeepSpeed-Kernels
  ☆68Updated last month
• tanyuqian / redco
  NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
  ☆65Updated 5 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 9 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 9 months ago
• 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
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 7 months ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 10 months ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆264Updated last week
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆70Updated last week
• NVIDIA / compute-eval
  Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…
  ☆38Updated 2 weeks ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 2 months ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆140Updated 6 months ago
• mayank31398 / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆40Updated this week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆131Updated last year
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆121Updated last month