ROCm / apexLinks
A PyTorch Extension:  Tools for easy mixed precision and distributed training in Pytorch
☆23Updated 3 weeks ago
Alternatives and similar repositories for apex
Users that are interested in apex are comparing it to the libraries listed below
Sorting:
- SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs☆44Updated this week
- RCCL Performance Benchmark Tests☆76Updated last week
- OpenAI Triton backend for Intel® GPUs☆215Updated this week
- ☆51Updated this week
- oneCCL Bindings for Pytorch*☆102Updated 2 months ago
- Development repository for the Triton language and compiler☆136Updated 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 4 months ago
- ☆62Updated 10 months ago
- Ahead of Time (AOT) Triton Math Library☆80Updated last week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆147Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆107Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆388Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆154Updated this week
- ROCm Communication Collectives Library (RCCL)☆390Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated this week
- Provides the examples to write and build Habana custom kernels using the HabanaTools☆23Updated 6 months ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆254Updated this week
- rocWMMA☆136Updated this week
- AMD's graph optimization engine.☆262Updated this week
- rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.☆121Updated last week
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆41Updated last year
- Bandwidth test for ROCm☆67Updated this week
- ☆27Updated last month
- ☆157Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆29Updated this week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆478Updated last week
- oneAPI Collective Communications Library (oneCCL)☆245Updated last week
- An extension library of WMMA API (Tensor Core API)☆107Updated last year
- A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high…☆84Updated this week
- ☆109Updated last year