HFAiLab / pytorch_distributed
The test of different distributed-training methods on High-Flyer AIHPC
☆24Updated 2 years ago
Alternatives and similar repositories for pytorch_distributed:
Users that are interested in pytorch_distributed are comparing it to the libraries listed below
- Benchmark tests supporting the TiledCUDA library.☆15Updated 4 months ago
- ☆11Updated last year
- ☆30Updated 10 months ago
- ☆52Updated this week
- Odysseus: Playground of LLM Sequence Parallelism☆68Updated 9 months ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- differentiable top-k operator☆21Updated 3 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- study of cutlass☆21Updated 4 months ago
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 3 years ago
- Implementation of IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).☆22Updated 9 months ago
- An object detection codebase based on MegEngine.☆28Updated 2 years ago
- 方便扩展的Cuda算子理解和优化框架,仅用在学习使用☆13Updated 9 months ago
- Efficient Mixture of Experts for LLM Paper List☆47Updated 3 months ago
- ☆15Updated last year
- An external memory allocator example for PyTorch.☆14Updated 3 years ago
- A small framework mimics PyTorch using CuPy or NumPy☆27Updated 3 years ago
- ☆18Updated 2 years ago
- ☆11Updated last year
- ☆22Updated last year
- ☆17Updated last year
- ☆17Updated last year
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆61Updated last year
- GPTQ inference TVM kernel☆38Updated 11 months ago
- Datasets, Transforms and Models specific to Computer Vision☆85Updated last year
- OneFlow Serving☆20Updated 3 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆35Updated 3 weeks ago
- Quantized Attention on GPU☆45Updated 4 months ago
- An industrial extension library of pytorch to accelerate large scale model training☆29Updated last month
- ☆19Updated 3 weeks ago