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:
- ☆51Updated this week
- SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs☆51Updated this week
- RCCL Performance Benchmark Tests☆79Updated this week
- Development repository for the Triton language and compiler☆137Updated last week
- Ahead of Time (AOT) Triton Math Library☆83Updated last week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆255Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆108Updated this week
- ROCm Communication Collectives Library (RCCL)☆399Updated this week
- oneCCL Bindings for Pytorch* (deprecated)☆102Updated 2 weeks ago
- Bandwidth test for ROCm☆69Updated this week
- OpenAI Triton backend for Intel® GPUs☆219Updated this week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆483Updated this week
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆41Updated last year
- rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.☆126Updated 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
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆147Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆114Updated this week
- ☆27Updated last month
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated 3 weeks ago
- AMD's graph optimization engine.☆266Updated this week
- Assembler for NVIDIA Volta and Turing GPUs☆233Updated 3 years ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆137Updated this week
- A tool for generating information about the matrix multiplication instructions in AMD Radeon™ and AMD Instinct™ accelerators☆120Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆165Updated last week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆388Updated this week
- Experimental projects related to TensorRT☆114Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆362Updated this week
- Provides the examples to write and build Habana custom kernels using the HabanaTools☆23Updated 7 months ago
- ☆62Updated 11 months ago
- monorepo for rocm libraries☆177Updated this week