ROCm / apexLinks
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
☆23Updated last week
Alternatives and similar repositories for apex
Users that are interested in apex are comparing it to the libraries listed below
Sorting:
- RCCL Performance Benchmark Tests☆77Updated this week
- ☆47Updated this week
- SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs☆40Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆103Updated this week
- OpenAI Triton backend for Intel® GPUs☆210Updated this week
- Development repository for the Triton language and compiler☆133Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆249Updated this week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆63Updated 3 months ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated last week
- ☆63Updated 9 months ago
- Provides the examples to write and build Habana custom kernels using the HabanaTools☆22Updated 5 months ago
- rocWMMA☆133Updated last week
- ROCm Communication Collectives Library (RCCL)☆386Updated this week
- oneAPI Collective Communications Library (oneCCL)☆245Updated 2 weeks ago
- Ahead of Time (AOT) Triton Math Library☆77Updated this week
- oneCCL Bindings for Pytorch*☆102Updated 2 months ago
- rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.☆119Updated this week
- Bandwidth test for ROCm☆66Updated last week
- ☆27Updated last week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆152Updated last week
- AMD's graph optimization engine.☆253Updated this week
- Example of using pytorch's open device registration API☆30Updated 2 years ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆161Updated last week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆470Updated this week
- ☆108Updated last year
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆111Updated this week
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆41Updated last year
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆383Updated last week
- An extension library of WMMA API (Tensor Core API)☆106Updated last year
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆149Updated last week