cisco-open / pymultiworldLinks
A framework for PyTorch to enable fault management for collective communication libraries (CCL) such as NCCL
☆20Updated this week
Alternatives and similar repositories for pymultiworld
Users that are interested in pymultiworld are comparing it to the libraries listed below
Sorting:
- Ship correct and fast LLM kernels to PyTorch☆126Updated this week
- JAX backend for SGL☆200Updated this week
- torchcomms: a modern PyTorch communications API☆309Updated this week
- extensible collectives library in triton☆91Updated 8 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆253Updated 2 months ago
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆256Updated last week
- Accelerating MoE with IO and Tile-aware Optimizations☆351Updated this week
- A resilient distributed training framework☆96Updated last year
- Triton-based Symmetric Memory operators and examples☆67Updated 2 months ago
- 🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.☆64Updated 10 months ago
- Applied AI experiments and examples for PyTorch☆311Updated 3 months ago
- ☆115Updated last year
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆71Updated this week
- Allow torch tensor memory to be released and resumed later☆187Updated 2 weeks ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆162Updated 3 months ago
- ring-attention experiments☆160Updated last year
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆84Updated last month
- Autonomous GPU Kernel Generation via Deep Agents☆187Updated this week
- Toolchain built around the Megatron-LM for Distributed Training☆80Updated 2 weeks ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆81Updated last year
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆216Updated last week
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆219Updated last year
- ☆252Updated last year
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆271Updated 3 weeks ago
- A NCCL extension library, designed to efficiently offload GPU memory allocated by the NCCL communication library.☆66Updated this week
- Genai-bench is a powerful benchmark tool designed for comprehensive token-level performance evaluation of large language model (LLM) serv…☆247Updated last week
- Collection of kernels written in Triton language☆172Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆135Updated 6 months ago
- ☆94Updated 3 years ago
- [OSDI'24] Serving LLM-based Applications Efficiently with Semantic Variable☆203Updated last year