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 2 months ago
• open-lm-engine / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆46Updated this week
• ScalingIntelligence / hydragen
  Hydragen: High-Throughput LLM Inference with Shared Prefixes
  ☆36Updated last year
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆44Updated this week
• Jokeren / triton-samples
  ☆28Updated 4 months ago
• simveit / effective_transpose
  Effective transpose on Hopper GPU
  ☆20Updated last month
• pytorch-labs / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆153Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• Dao-AILab / grouped-latent-attention
  ☆93Updated last week
• triton-lang / kernels
  ☆80Updated 6 months ago
• spcl / sten
  Sparsity support for PyTorch
  ☆35Updated 2 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.
  ☆44Updated 10 months ago
• mayank31398 / ladder-residual-inference
  ☆13Updated 3 weeks ago
• mlcommons / logging
  MLPerf™ logging library
  ☆36Updated last month
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆93Updated last week
• stanford-futuredata / stk
  ☆105Updated 9 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆125Updated 2 months ago
• NVIDIA / jaxpp
  JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training
  ☆45Updated 2 weeks ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆223Updated 10 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆116Updated 6 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆127Updated this week
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 11 months ago
• MekkCyber / CutlassAcademy
  A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS
  ☆181Updated last month
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆133Updated last year
• determined-ai / determined-examples
  Example ML projects that use the Determined library.
  ☆32Updated 8 months ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆99Updated last year
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆74Updated 3 weeks ago
• deepspeedai / DeepSpeed-Kernels
  ☆71Updated 2 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆217Updated 6 months ago