ROCm / apexLinks
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
☆23Updated 2 weeks ago
Alternatives and similar repositories for apex
Users that are interested in apex are comparing it to the libraries listed below
Sorting:
- ☆41Updated last week
- OpenAI Triton backend for Intel® GPUs☆205Updated this week
- RCCL Performance Benchmark Tests☆73Updated last week
- A CUTLASS implementation using SYCL☆35Updated last week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆62Updated 2 months ago
- oneCCL Bindings for Pytorch*☆100Updated 3 weeks ago
- Development repository for the Triton language and compiler☆127Updated this week
- ROCm Communication Collectives Library (RCCL)☆360Updated this week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆449Updated this week
- Ahead of Time (AOT) Triton Math Library☆75Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆92Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆247Updated 2 weeks ago
- ☆62Updated 8 months ago
- Bandwidth test for ROCm☆65Updated this week
- Example of using pytorch's open device registration API☆30Updated 2 years ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated 2 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆111Updated this week
- Experimental projects related to TensorRT☆111Updated last week
- ☆27Updated 2 months ago
- AMD's graph optimization engine.☆240Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆349Updated this week
- Provides the examples to write and build Habana custom kernels using the HabanaTools☆22Updated 4 months ago
- A tool for bandwidth measurements on NVIDIA GPUs.☆517Updated 4 months ago
- Fast and memory-efficient exact attention☆183Updated 2 weeks ago
- oneAPI Collective Communications Library (oneCCL)☆242Updated 3 weeks ago
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆40Updated last year
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆146Updated 2 weeks ago
- Assembler for NVIDIA Volta and Turing GPUs☆229Updated 3 years ago
- A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high…☆77Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆382Updated this week