ROCm / jaxLinks
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
☆24Updated this week
Alternatives and similar repositories for jax
Users that are interested in jax are comparing it to the libraries listed below
Sorting:
- 8-bit CUDA functions for PyTorch☆66Updated last month
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆23Updated 3 weeks ago
- The AMD rocAL is designed to efficiently decode and process images and videos from a variety of storage formats and modify them through a…☆21Updated this week
- CMake modules used within the ROCm libraries☆67Updated this week
- ROCm's Thunk Interface☆91Updated 7 months ago
- Development repository for the Triton language and compiler☆136Updated this week
- RCCL Performance Benchmark Tests☆76Updated last week
- A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high…☆84Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆147Updated this week
- ☆152Updated last week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆114Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆388Updated this week
- Bandwidth test for ROCm☆67Updated this week
- ☆67Updated this week
- ☆61Updated 2 years ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated this week
- ☆126Updated this week
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆265Updated this week
- A collection of examples for the ROCm software stack☆250Updated this week
- ☆51Updated 5 months ago
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆33Updated last week
- ROCm Communication Collectives Library (RCCL)☆390Updated this week
- OpenAI Triton backend for Intel® GPUs☆215Updated this week
- This is ROCgdb, the ROCm source-level debugger for Linux, based on GDB, the GNU source-level debugger.☆63Updated this week
- rocWMMA☆136Updated this week
- oneAPI Level Zero Conformance & Performance test content☆57Updated last week
- ☆137Updated 3 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆198Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆107Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆69Updated 2 weeks ago