HFAiLab / pytorch_distributedLinks
The test of different distributed-training methods on High-Flyer AIHPC
☆26Updated 3 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:
- Datasets, Transforms and Models specific to Computer Vision☆90Updated 2 years ago
- ☆18Updated 3 years ago
- ☆12Updated 2 years ago
- study of cutlass☆22Updated last year
- ☆19Updated 3 years ago
- SuperDebug,debug如此简单!☆17Updated 3 years ago
- HFAI deep learning models☆156Updated 2 years ago
- SGEMM optimization with cuda step by step☆21Updated last year
- CVFusion is an open-source deep learning compiler to fuse the OpenCV operators.☆33Updated 3 years ago
- An object detection codebase based on MegEngine.☆28Updated 3 years ago
- Tutorials for writing high-performance GPU operators in AI frameworks.☆134Updated 2 years ago
- differentiable top-k operator☆22Updated last year
- [AAAI2023] NAS-LID: Efficient Neural Architecture Search with Local Intrinsic Dimension☆17Updated 3 years ago
- CUDA 6大并行计算模式 代码与笔记☆61Updated 5 years ago
- Benchmark tests supporting the TiledCUDA library.☆18Updated last year
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated 2 years ago
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 4 years ago
- Self Reproduction Code of Paper "Reducing Transformer Key-Value Cache Size with Cross-Layer Attention (MIT CSAIL)☆18Updated last year
- 各种深度学习(DL)框架分布式训练,包括:Tensorflow、Tensorflow2、Pytorch、Chainer、Caffe、Mxnet ...☆22Updated 5 years ago
- 跨平台的容器化Linux桌面环境☆73Updated 10 months ago
- ☆18Updated 2 years ago
- Tutorials to GPU programming. Reading notes.☆18Updated 2 years ago
- FireFlyer Record file format, writer and reader for DL training samples.☆238Updated 3 years ago
- [CVPRW 2021] Dynamic-OFA: Runtime DNN Architecture Switching for Performance Scaling on Heterogeneous Embedded Platforms☆30Updated 3 years ago
- An external memory allocator example for PyTorch.☆16Updated 5 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆57Updated 2 years ago
- Examples and instructions about use LLMs (especially ChatGPT) for PhD☆107Updated 2 years ago
- A high-performance distributed deep learning system targeting large-scale and automated distributed training. If you have any interests, …☆123Updated 2 years ago
- GPTQ inference TVM kernel☆41Updated last year
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆133Updated 2 years ago