Azure / MS-AMP-Examples

Related projects ⓘ

Alternatives and complementary repositories for MS-AMP-Examples

• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆279Updated last week
• RulinShao / LightSeq
  Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers
  ☆196Updated 2 months ago
• stanford-futuredata / stk
  ☆88Updated 2 months ago
• zhuohan123 / terapipe
  ☆65Updated 3 years ago
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆67Updated 3 months ago
• xuqifan897 / Optimus
  ☆26Updated 3 years ago
• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆217Updated 5 months ago
• Azure / MS-AMP
  Microsoft Automatic Mixed Precision Library
  ☆523Updated last month
• anyscale / llm-continuous-batching-benchmarks
  ☆109Updated 7 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆156Updated last year
• microsoft / nnscaler
  nnScaler: Compiling DNN models for Parallel Training
  ☆64Updated 2 weeks ago
• Guangxuan-Xiao / torch-int
  This repository contains integer operators on GPUs for PyTorch.
  ☆181Updated last year
• HPDL-Group / Merak
  ☆74Updated 3 weeks ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated last week
• cli99 / flops-profiler
  pytorch-profiler
  ☆49Updated last year
• ColfaxResearch / cutlass-kernels
  ☆162Updated 4 months ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆34Updated 2 years ago
• pytorch-tpu / llama
  Inference code for LLaMA models
  ☆19Updated 5 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆96Updated this week
• spcl / substation
  Research and development for optimizing transformers
  ☆124Updated 3 years ago
• thu-pacman / FasterMoE
  ☆70Updated 2 years ago
• microsoft / vattention
  Dynamic Memory Management for Serving LLMs without PagedAttention
  ☆228Updated last week
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆196Updated 2 weeks ago
• saareliad / FTPipe
  FTPipe and related pipeline model parallelism research.
  ☆41Updated last year
• awslabs / raf
  ☆140Updated last year
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆75Updated last week
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆160Updated last week
• feifeibear / long-context-attention
  USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference
  ☆353Updated last week
• awslabs / slapo
  A schedule language for large model training
  ☆141Updated 4 months ago
• bytedance / flux
  A fast communication-overlapping library for tensor parallelism on GPUs.
  ☆219Updated 2 weeks ago