ROCm / apexLinks
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
☆25Updated this week
Alternatives and similar repositories for apex
Users that are interested in apex are comparing it to the libraries listed below
Sorting:
- ☆60Updated last week
- Development repository for the Triton language and compiler☆140Updated last week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆86Updated 2 weeks ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆114Updated this week
- SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs☆66Updated last week
- OpenAI Triton backend for Intel® GPUs☆226Updated this week
- Ahead of Time (AOT) Triton Math Library☆88Updated last week
- ☆61Updated last year
- Fast and memory-efficient exact attention☆214Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆410Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo. NOTE: develop branch is maintained as a read-only mirror☆518Updated this week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆64Updated 7 months ago
- ☆29Updated this week
- oneCCL Bindings for Pytorch* (deprecated)☆104Updated last month
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆144Updated last week
- An extension library of WMMA API (Tensor Core API)☆109Updated last year
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆138Updated 2 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆148Updated 2 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆254Updated 2 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated 2 weeks ago
- ☆23Updated last week
- This repository contains the results and code for the MLPerf™ Training v2.0 benchmark.☆29Updated last year
- ☆167Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆390Updated this week
- PArallelLOOPgEneratoR: Threaded Loops Code Generation Infrastructure targeting Tensor Contraction Applications such as GEMMs, Convolution…☆19Updated 2 weeks ago
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆41Updated last year
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆113Updated this week
- Provides the examples to write and build Habana custom kernels using the HabanaTools☆25Updated 9 months ago
- Example of using pytorch's open device registration API☆31Updated 3 years ago
- Bandwidth test for ROCm☆75Updated last week