AMD ROCm™ Software

rocMLIR

hipBLASLt

hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditional BLAS library

xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

ROCm

AMD ROCm™ Software - GitHub Home

composable_kernel

Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators

ROCR-Runtime

ROCm Platform Runtime: ROCr a HPC market enhanced HSA based runtime

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

aomp

AOMP is an open source Clang/LLVM based compiler with added support for the OpenMP® API on Radeon™ GPUs. Use this repository for releases, issues, documentation, packaging, and examples.

migraphx-benchmark

MIVisionX

MIVisionX toolkit is a set of comprehensive computer vision and machine intelligence libraries, utilities, and applications bundled into a single toolkit. AMD MIVisionX also delivers a highly optimized open-source implementation of the Khronos OpenVX™ and OpenVX™ Extensions.

AMDMIGraphX

AMD's graph optimization engine.

hipify_torch

llvm-project

This is the AMD-maintained fork of the LLVM git repository. This repository accepts pull requests and issues related to AMD fork-specific topics (amd/*). For all other issues/PRs, please submit upstream at https://github.com/llvm/llvm-project.

tensorflow-upstream

TensorFlow ROCm port

rocAL

The AMD rocAL is designed to efficiently decode and process images and videos from a variety of storage formats and modify them through a processing graph programmable by the user.

Megatron-LM

Ongoing research training transformer models at scale

rccl

ROCm Communication Collectives Library (RCCL)

rocSOLVER

Next generation LAPACK implementation for ROCm platform

MIOpen

AMD's Machine Intelligence Library

Tensile

Stretching GPU performance for GEMMs and tensor contractions.

rpp

AMD ROCm Performance Primitives (RPP) library is a comprehensive high-performance computer vision library for AMD processors with HIP/OpenCL/CPU back-ends.

horovod

Distributed training framework for TensorFlow, Keras, and PyTorch.

aomp-extras

hostcall services library, math library, and utilities

vllm

A high-throughput and memory-efficient inference and serving engine for LLMs

hipSOLVER

ROCm SOLVER marshalling library

hipSPARSE

ROCm SPARSE marshalling library

hipRAND

Random number library that generate pseudo-random and quasi-random numbers.

rocRAND

RAND library for HIP programming language

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

triton

Development repository for the Triton language and compiler