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
  ☆83Updated 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.
  ☆43Updated 7 months ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆91Updated 9 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆104Updated 2 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated 11 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆75Updated last year
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 4 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆105Updated this week
• unixpickle / learn-ptx
  Learning about CUDA by writing PTX code.
  ☆35Updated 11 months ago
• deepspeedai / DeepSpeed-Kernels
  ☆59Updated 2 weeks ago
• Jokeren / triton-samples
  ☆25Updated last month
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆89Updated this week
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆168Updated last month
• stanford-futuredata / stk
  ☆100Updated 5 months ago
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆76Updated last month
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆20Updated 4 years ago
• triton-lang / kernels
  ☆67Updated 3 months ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 3 months ago
• KernelTuner / kernel_tuner_tutorial
  A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/
  ☆30Updated 5 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆221Updated 6 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆180Updated this week
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆64Updated 2 years ago
• spcl / sten
  Sparsity support for PyTorch
  ☆33Updated last week
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆225Updated this week
• olcf / NVIDIA-tensor-core-examples
  ☆17Updated 5 years ago
• DS3Lab / CocktailSGD
  ☆26Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆43Updated last week
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆46Updated 2 months ago
• UmerHA / triton_util
  Make triton easier
  ☆44Updated 8 months ago
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆173Updated last week