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
Sorting:
- ☆11Updated last year
- An object detection codebase based on MegEngine.☆28Updated 2 years ago
- differentiable top-k operator☆21Updated 4 months ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆18Updated last week
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- A simple program scheduler for your code on different devices.☆11Updated 9 months ago
- ☆30Updated 11 months ago
- ☆17Updated last year
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆62Updated last year
- ☆70Updated last week
- ☆11Updated 2 years ago
- An external memory allocator example for PyTorch.☆14Updated 3 years ago
- Benchmarking Attention Mechanism in Vision Transformers.☆18Updated 2 years ago
- TVMScript kernel for deformable attention☆25Updated 3 years ago
- ☆38Updated last year
- Benchmark tests supporting the TiledCUDA library.☆16Updated 5 months ago
- [ICLR 2024] This is the official PyTorch implementation of "QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Mod…☆36Updated last year
- study of cutlass☆21Updated 6 months ago
- Codes For Sharing☆38Updated 4 years ago
- Efficient Mixture of Experts for LLM Paper List☆65Updated 5 months ago
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆41Updated 2 weeks ago
- Training LLaMA language model with MMEngine! It supports LoRA fine-tuning!☆40Updated 2 years ago
- Datasets, Transforms and Models specific to Computer Vision☆84Updated last year
- ☆13Updated 3 years ago
- ☆16Updated 2 years ago
- ☆24Updated 2 years ago
- Paddle Automatically Diff Precision Toolkits.☆49Updated last year
- [CVPR-2023] Towards Any Structural Pruning☆16Updated 2 years ago
- ☆16Updated last year
- ☆21Updated 4 years ago