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
  ☆87Updated 3 months ago
• open-lm-engine / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆55Updated this week
• Jokeren / triton-samples
  ☆28Updated 6 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆46Updated 2 weeks ago
• HabanaAI / Megatron-DeepSpeed
  Intel Gaudi's Megatron DeepSpeed Large Language Models for training
  ☆13Updated 6 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.
  ☆46Updated last year
• deepspeedai / DeepSpeed-Kernels
  ☆74Updated 3 months ago
• spcl / sten
  Sparsity support for PyTorch
  ☆35Updated 3 months ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated last year
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆158Updated 3 weeks ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆143Updated 9 months ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆99Updated last year
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆187Updated this week
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated 11 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆116Updated 7 months ago
• HazyResearch / train-tk
  train with kittens!
  ☆61Updated 8 months ago
• simveit / effective_transpose
  Effective transpose on Hopper GPU
  ☆23Updated 2 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• lianakoleva / no-libtorch-compile
  ☆21Updated 4 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆19Updated 11 months ago
• stanford-futuredata / stk
  ☆106Updated 10 months ago
• triton-lang / kernels
  ☆83Updated 8 months ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆49Updated 11 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆125Updated this week
• fw-ai / llama-cuda-graph-example
  Example of applying CUDA graphs to LLaMA-v2
  ☆12Updated last year
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆31Updated 5 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆136Updated 3 months ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆206Updated this week
• ScalingIntelligence / hydragen
  Hydragen: High-Throughput LLM Inference with Shared Prefixes
  ☆40Updated last year