ROCm / apexLinks
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
☆24Updated last week
Alternatives and similar repositories for apex
Users that are interested in apex are comparing it to the libraries listed below
Sorting:
- RCCL Performance Benchmark Tests☆81Updated last week
- ☆52Updated this week
- OpenAI Triton backend for Intel® GPUs☆222Updated this week
- Development repository for the Triton language and compiler☆137Updated this week
- SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs☆59Updated this week
- Ahead of Time (AOT) Triton Math Library☆84Updated last month
- A high-throughput and memory-efficient inference and serving engine for LLMs☆111Updated this week
- Fast and memory-efficient exact attention☆202Updated this week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆496Updated this week
- oneCCL Bindings for Pytorch* (deprecated)☆103Updated last month
- ☆27Updated 2 months ago
- ROCm Communication Collectives Library (RCCL)☆404Updated this week
- A tool for bandwidth measurements on NVIDIA GPUs.☆580Updated 7 months ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated this week
- ☆62Updated 11 months ago
- Intel® Tensor Processing Primitives extension for Pytorch*☆17Updated last week
- Experimental projects related to TensorRT☆117Updated this week
- ☆22Updated last month
- ☆248Updated last year
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆63Updated 5 months ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆256Updated last week
- ☆159Updated 7 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆363Updated last week
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆121Updated this week
- QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.☆36Updated 3 months ago
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆41Updated last year
- AI Tensor Engine for ROCm☆311Updated this week
- rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.☆131Updated this week
- Provides the examples to write and build Habana custom kernels using the HabanaTools☆24Updated 7 months ago
- ☆150Updated 11 months ago