ROCm / apexLinks
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
☆25Updated 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☆63Updated this week
- ☆55Updated this week
- RCCL Performance Benchmark Tests☆85Updated last month
- Ahead of Time (AOT) Triton Math Library☆85Updated 3 weeks ago
- OpenAI Triton backend for Intel® GPUs☆223Updated this week
- Development repository for the Triton language and compiler☆140Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆113Updated this week
- ROCm Communication Collectives Library (RCCL)☆409Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆253Updated last week
- oneCCL Bindings for Pytorch* (deprecated)☆104Updated last week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆505Updated this week
- ☆28Updated 3 months ago
- Bandwidth test for ROCm☆73Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆391Updated last week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆63Updated 6 months ago
- Experimental projects related to TensorRT☆118Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆148Updated this week
- AMD's graph optimization engine.☆269Updated this week
- Fast and memory-efficient exact attention☆207Updated this week
- A tool for bandwidth measurements on NVIDIA GPUs.☆602Updated 8 months ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated last month
- ☆61Updated last year
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆368Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆12Updated last year
- rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.☆140Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆29Updated last week
- AI Tensor Engine for ROCm☆330Updated last week
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆41Updated last year
- A collection of examples for the ROCm software stack☆266Updated last week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆154Updated last month