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:
- A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high…☆73Updated this week
- 8-bit CUDA functions for PyTorch☆53Updated this week
- Bandwidth test for ROCm☆58Updated last month
- CMake modules used within the ROCm libraries☆67Updated this week
- ☆47Updated 3 weeks ago
- Ongoing research training transformer models at scale☆23Updated 2 weeks ago
- TransferBench is a utility capable of benchmarking simultaneous copies between user-specified devices (CPUs/GPUs)☆41Updated last week
- ☆20Updated 3 months ago
- ☆36Updated this week
- AMD SMI☆71Updated this week
- ROCm's Thunk Interface☆91Updated 3 months 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…☆19Updated this week
- Development repository for the Triton language and compiler☆125Updated this week
- ROCm Communication Collectives Library (RCCL)☆342Updated this week
- HPCG benchmark based on ROCm platform☆37Updated 3 weeks ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆22Updated 2 weeks ago
- A collection of examples for the ROCm software stack☆224Updated this week
- ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs☆83Updated last week
- ☆25Updated this week
- ROCm Documentation Python package for ReadTheDocs build standardization☆16Updated last week
- This is ROCgdb, the ROCm source-level debugger for Linux, based on GDB, the GNU source-level debugger.☆59Updated last week
- Fast and memory-efficient exact attention☆174Updated this week
- ☆46Updated this week
- RCCL Performance Benchmark Tests☆68Updated last month
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆106Updated this week
- ROCm SPARSE marshalling library☆67Updated this week
- oneAPI Level Zero Conformance & Performance test content☆54Updated last week
- ROCm Device Libraries☆97Updated last year
- ☆38Updated this week
- rocWMMA☆115Updated last week