axonn-ai / axonn

Related projects ⓘ

Alternatives and complementary repositories for axonn

• cchan / tccl
  extensible collectives library in triton
  ☆72Updated last month
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆98Updated last week
• ScalingIntelligence / hydragen
  Hydragen: High-Throughput LLM Inference with Shared Prefixes
  ☆24Updated 6 months ago
• spcl / sten
  Sparsity support for PyTorch
  ☆31Updated 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 …
  ☆63Updated 2 years ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆60Updated 8 months ago
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆19Updated 4 years ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆166Updated 3 weeks ago
• hpcgroup / pipit
  Analyze parallel execution traces using pandas dataframes
  ☆24Updated 2 weeks ago
• NERSC / sc22-dl-tutorial
  Material for the SC22 Deep Learning at Scale Tutorial
  ☆39Updated last year
• triton-lang / kernels
  ☆45Updated 2 weeks ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆68Updated 3 weeks ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆34Updated 2 years ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆111Updated 5 months ago
• LLNL / hatchet
  Graph-indexed Pandas DataFrames for analyzing hierarchical performance data
  ☆30Updated 3 weeks ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆134Updated 2 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated 3 weeks ago
• hao-ai-lab / MuxServe
  ☆46Updated 5 months ago
• NVIDIA / LDDL
  Distributed preprocessing and data loading for language datasets
  ☆39Updated 7 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆211Updated 3 months ago
• Jokeren / triton-samples
  ☆23Updated 2 months ago
• olcf / NVIDIA-tensor-core-examples
  ☆15Updated 5 years ago
• yifuwang / symm-mem-recipes
  ☆12Updated last month
• mlcommons / logging
  MLPerf™ logging library
  ☆30Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆153Updated this week
• stanford-futuredata / stk
  ☆88Updated 2 months ago
• unixpickle / learn-ptx
  Learning about CUDA by writing PTX code.
  ☆28Updated 8 months ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆30Updated 3 months ago
• manishucsd / py-codegen
  ☆14Updated last month