HFAiLab / pytorch_distributed
The test of different distributed-training methods on High-Flyer AIHPC
☆22Updated 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
- FireFlyer Record file format, writer and reader for DL training samples.☆121Updated 2 years ago
- ☆16Updated 2 years ago
- An object detection codebase based on MegEngine.☆28Updated 2 years ago
- HFAI deep learning models☆99Updated last year
- ☆31Updated 7 months ago
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- ☆11Updated last year
- NeurIPS文章+代码(最新2021更新)☆22Updated 3 years ago
- differentiable top-k operator☆21Updated 3 weeks ago
- A small framework mimics PyTorch using CuPy or NumPy☆27Updated 2 years ago
- ☆15Updated last year
- An optimized version of alphafold 2 based on openfold☆12Updated last year
- ☆17Updated last year
- study of cutlass☆19Updated 2 months ago
- TVMScript kernel for deformable attention☆24Updated 3 years ago
- 2nd place solution of ECCV 2020 workshop VIPriors Image Classification Challenge, https://arxiv.org/abs/2008.00261☆13Updated 3 years ago
- Moved to https://github.com/NUS-HPC-AI-Lab/InfoBatch☆6Updated last year
- Datasets, Transforms and Models specific to Computer Vision☆84Updated last year
- ☆99Updated 3 years ago
- pytorch-profiler☆50Updated last year
- [ICLR 2024] This is the official PyTorch implementation of "QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Mod…☆35Updated 10 months ago
- 一个mmcv 的logger hook, 可以用来把模型结果推送到微信上☆20Updated 2 years ago
- A simple program scheduler for your code on different devices.☆11Updated 5 months ago
- CVFusion is an open-source deep learning compiler to fuse the OpenCV operators.☆28Updated 2 years ago
- [AAAI2023] NAS-LID: Efficient Neural Architecture Search with Local Intrinsic Dimension☆16Updated 2 years ago
- PyTorch Dataset Rank Dataset☆41Updated 3 years ago
- PyTorch code for Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆36Updated 4 months ago
- Distributed DataLoader For Pytorch Based On Ray☆24Updated 3 years ago
- ☆27Updated last month