Alternatives and similar repositories for TensorNVMe

Users that are interested in TensorNVMe are comparing it to the libraries listed below

• stanford-futuredata / stk
  ☆110Updated last year
• deepspeedai / DeepSpeed-Kernels
  ☆74Updated 5 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆116Updated 3 months ago
• triton-lang / kernels
  ☆88Updated 10 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆159Updated 2 years ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆263Updated last month
• cchan / tccl
  extensible collectives library in triton
  ☆87Updated 5 months ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆161Updated 2 months ago
• spcl / substation
  Research and development for optimizing transformers
  ☆129Updated 4 years ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆77Updated last year
• zhuohan123 / terapipe
  ☆75Updated 4 years ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆114Updated last year
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆79Updated 9 months ago
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆216Updated last year
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆81Updated last week
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆97Updated 2 months ago
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆221Updated 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 …
  ☆65Updated 3 years ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆48Updated last year
• AlibabaResearch / flash-llm
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆219Updated last year
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆82Updated last year
• exists-forall / striped_attention
  ☆40Updated last year
• anyscale / llm-continuous-batching-benchmarks
  ☆121Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆209Updated last week
• AlibabaPAI / FLASHNN
  ☆98Updated last year
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆40Updated 2 years ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆70Updated 4 months ago
• InternLM / turbomind
  ☆94Updated 5 months ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆134Updated 3 years ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated last year