Alternatives and similar repositories for torch_utils

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

• tile-ai / AttentionEngine
  ☆49Updated last month
• microsoft / AttentionEngine
  ☆74Updated last month
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆70Updated last year
• yanring / Megatron-MoE-ModelZoo
  Best practices for testing advanced Mixtral, DeepSeek, and Qwen series MoE models using Megatron Core MoE.
  ☆29Updated last month
• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆44Updated 7 months ago
• flashinfer-ai / cutlass-viz
  ☆60Updated 2 months ago
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆42Updated last week
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆69Updated 2 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆101Updated last month
• microsoft / chunk-attention
  ☆77Updated 2 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆47Updated 11 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆79Updated last week
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆39Updated 4 months ago
• InternLM / turbomind
  ☆87Updated 3 months ago
• dame-cell / Triformer
  Transformers components but in Triton
  ☆34Updated 2 months ago
• AlibabaPAI / FLASHNN
  ☆96Updated 10 months ago
• thunlp / TritonBench
  TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators
  ☆62Updated last month
• ISEEKYAN / mbridge
  ☆50Updated 2 weeks ago
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆87Updated 6 months ago
• feifeibear / DPSKV3MFU
  Estimate MFU for DeepSeekV3
  ☆25Updated 6 months ago
• ModelTC / awesome-lm-system
  Summary of system papers/frameworks/codes/tools on training or serving large model
  ☆57Updated last year
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆40Updated last year
• Dao-AILab / grouped-latent-attention
  ☆116Updated last month
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆184Updated this week
• Bruce-Lee-LY / decoding_attention
  Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.
  ☆38Updated last month
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆118Updated last month
• sail-sg / VocabularyParallelism
  Vocabulary Parallelism
  ☆19Updated 4 months ago
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆100Updated 3 months ago
• hao-ai-lab / vllm-ltr
  [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank
  ☆49Updated 8 months ago
• stanford-futuredata / stk
  ☆106Updated 10 months ago