Alternatives and similar repositories for torch_memory_saver

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

• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆209Updated 8 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆69Updated 10 months ago
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆121Updated 4 months ago
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆103Updated this week
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆78Updated 5 months ago
• flashinfer-ai / cutlass-viz
  ☆58Updated 3 weeks ago
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆41Updated 2 weeks ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆89Updated 2 weeks ago
• feifeibear / DPSKV3MFU
  Estimate MFU for DeepSeekV3
  ☆24Updated 4 months ago
• ByteDance-Seed / ByteCheckpoint
  ByteCheckpoint: An Unified Checkpointing Library for LFMs
  ☆209Updated last month
• NVlabs / COAT
  [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training
  ☆190Updated 3 weeks ago
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆38Updated 2 months ago
• hao-ai-lab / vllm-ltr
  [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank
  ☆48Updated 6 months ago
• InternLM / turbomind
  ☆84Updated last month
• thu-pacman / FasterMoE
  ☆82Updated 3 years ago
• stanford-futuredata / stk
  ☆104Updated 8 months ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆40Updated 9 months ago
• Infini-AI-Lab / MagicDec
  [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding
  ☆116Updated 5 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆124Updated this week
• madsys-dev / deepseekv2-profile
  ☆132Updated 2 months ago
• sail-sg / VocabularyParallelism
  Vocabulary Parallelism
  ☆19Updated 2 months ago
• triton-lang / kernels
  ☆79Updated 6 months ago
• MayDomine / Seq1F1B
  Sequence-level 1F1B schedule for LLMs.
  ☆17Updated 11 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆69Updated last week
• fla-org / flame
  🔥 A minimal training framework for scaling FLA models
  ☆128Updated last week
• thunlp / Ouroboros
  Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)
  ☆103Updated last month
• AlibabaPAI / FLASHNN
  ☆94Updated 8 months ago
• kwai / Megatron-Kwai
  [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…
  ☆54Updated 9 months ago
• mit-han-lab / Quest
  [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference
  ☆279Updated 5 months ago
• microsoft / AttentionEngine
  ☆70Updated last week