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 this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆86Updated 2 weeks ago
- Ahead of Time (AOT) Triton Math Library☆88Updated last week
- OpenAI Triton backend for Intel® GPUs☆226Updated this week
- ☆29Updated this week
- oneCCL Bindings for Pytorch* (deprecated)☆104Updated last month
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆113Updated this week
- SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs☆66Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆114Updated this week
- Development repository for the Triton language and compiler☆140Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆410Updated this week
- MAD (Model Automation and Dashboarding)☆31Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆255Updated this week
- AMD's graph optimization engine.☆275Updated this week
- Bandwidth test for ROCm☆75Updated last 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
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆29Updated last week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo. NOTE: develop branch is maintained as a read-only mirror☆518Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated 2 weeks ago
- ☆61Updated last year
- Fast and memory-efficient exact attention☆214Updated this week
- AI Tensor Engine for ROCm☆351Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆148Updated 2 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆144Updated last week
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆42Updated last year
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆12Updated last year
- An extension library of WMMA API (Tensor Core API)☆109Updated last year
- A tool for bandwidth measurements on NVIDIA GPUs.☆618Updated 9 months ago
- Intel® Tensor Processing Primitives extension for Pytorch*☆18Updated 3 weeks ago
- ☆23Updated last week