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
  ☆85Updated 2 weeks ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• spcl / sten
  Sparsity support for PyTorch
  ☆34Updated 3 weeks ago
• deepspeedai / DeepSpeed-Kernels
  ☆68Updated 3 weeks ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆92Updated 11 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆223Updated 8 months ago
• triton-lang / kernels
  ☆77Updated 5 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated last year
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆21Updated 4 years ago
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆90Updated 2 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆125Updated 4 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆113Updated 4 months ago
• CentML / DeepView.Profile
  🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.
  ☆60Updated 2 months ago
• ScalingIntelligence / hydragen
  Hydragen: High-Throughput LLM Inference with Shared Prefixes
  ☆36Updated 11 months ago
• NVIDIA / compute-eval
  Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…
  ☆15Updated 2 weeks ago
• stanford-futuredata / stk
  ☆103Updated 7 months ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆46Updated 4 months ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆115Updated last week
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆118Updated 2 weeks ago
• mayank31398 / cute-kernels
  A bunch of kernels that might make stuff slower 😉
  ☆31Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆199Updated this week
• cloneofsimo / min-fsdp
  ☆76Updated 9 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆126Updated 4 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
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆110Updated this week
• HabanaAI / Megatron-DeepSpeed
  Intel Gaudi's Megatron DeepSpeed Large Language Models for training
  ☆13Updated 4 months ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆131Updated last year
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆81Updated 5 months ago
• Jokeren / triton-samples
  ☆27Updated 3 months ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆38Updated 8 months ago