fzyzcjy / torch_utilsLinks
Utility scripts for PyTorch (e.g. Memory profiler that understands more low-level allocations such as NCCL)
☆55Updated 3 weeks ago
Alternatives and similar repositories for torch_utils
Users that are interested in torch_utils are comparing it to the libraries listed below
Sorting:
- PyTorch bindings for CUTLASS grouped GEMM.☆123Updated 4 months ago
- ☆98Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆152Updated last month
- ☆64Updated 5 months ago
- ☆42Updated last year
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆130Updated 2 weeks ago
- Allow torch tensor memory to be released and resumed later☆142Updated last week
- DeeperGEMM: crazy optimized version☆71Updated 4 months ago
- A simple calculation for LLM MFU.☆46Updated 3 weeks ago
- A lightweight design for computation-communication overlap.☆177Updated 2 weeks ago
- Implement Flash Attention using Cute.☆95Updated 9 months ago
- ☆95Updated 6 months ago
- nnScaler: Compiling DNN models for Parallel Training☆118Updated last week
- ☆78Updated 5 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆221Updated 2 years ago
- A collection of memory efficient attention operators implemented in the Triton language.☆279Updated last year
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆40Updated 7 months ago
- ☆75Updated 4 years ago
- ☆88Updated 3 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆50Updated last year
- ☆112Updated last month
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆60Updated last month
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆65Updated last year
- Estimate MFU for DeepSeekV3☆24Updated 8 months ago
- Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.☆98Updated 2 weeks ago
- ☆50Updated 4 months ago
- ☆147Updated 7 months ago
- Bridge Megatron-Core to Hugging Face/Reinforcement Learning☆130Updated last week
- ☆298Updated last week