Skip to content

@ROCm AMD ROCm™ Software

Change the repository type filter

All

    Repositories list

    288 repositories

    • xla

      Public
      A machine learning compiler for GPUs, CPUs, and ML accelerators
      C++
      •
      Apache License 2.0
      •427•2•0•13•Updated Oct 30, 2024Oct 30, 2024

    • HIP

      Public
      HIP: C++ Heterogeneous-Compute Interface for Portability
      hiphip-runtimehipifyhip-kernel-languagehip-portabilitycuda
      C++
      •
      MIT License
      •531•3.7k•35•49•Updated Oct 30, 2024Oct 30, 2024

    • rocMLIR

      Public
      C++
      •40•127•2•17•Updated Oct 30, 2024Oct 30, 2024

    • composable_kernel

      Public
      Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
      C++
      •
      Other
      •124•306•42•35•Updated Oct 30, 2024Oct 30, 2024

    • xformers

      Public
      Hackable and optimized Transformers building blocks, supporting a composable construction.
      Python
      •
      Other
      •610•20•8•0•Updated Oct 30, 2024Oct 30, 2024

    • AMDMIGraphX

      Public
      AMD's graph optimization engine.
      C++
      •
      MIT License
      •84•185•344•36•Updated Oct 30, 2024Oct 30, 2024

    • MIOpen

      Public
      AMD's Machine Intelligence Library
      Assembly
      •
      Other
      •224•1.1k•254•45•Updated Oct 30, 2024Oct 30, 2024

    • hipBench

      Public
      HIP Kernel Benchmarking Library
      C++
      •
      Apache License 2.0
      •1•4•0•0•Updated Oct 30, 2024Oct 30, 2024

    • rocPRIM

      Public
      ROCm Parallel Primitives
      amdgpuparallelcudaprimitivehiprocm
      C++
      •
      MIT License
      •69•160•1•12•Updated Oct 30, 2024Oct 30, 2024

    • omnitrace

      Public
      Omnitrace: Application Profiling, Tracing, and Analysis
      pythonlinuxprofilertracingperformance-metricsperformance-analysisprofilingperformance-monitoringcode-coveragehardware-counters
      C++
      •
      MIT License
      •28•299•17•6•Updated Oct 30, 2024Oct 30, 2024

    • timemory

      Public
      MIT License
      •1•1•0•0•Updated Oct 30, 2024Oct 30, 2024

    • hipBLASLt

      Public
      hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditional BLAS library
      machine-learningamdassemblymatrix-multiplicationblashipgpu-computinggemmrocm
      Assembly
      •
      MIT License
      •85•57•7•64•Updated Oct 30, 2024Oct 30, 2024

    • llvm-project

      Public
      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.
      LLVM
      •
      Other
      •12k•116•25•18•Updated Oct 30, 2024Oct 30, 2024

    • hipSOLVER

      Public
      ROCm SOLVER marshalling library
      C++
      •
      MIT License
      •23•24•1•3•Updated Oct 30, 2024Oct 30, 2024

    • pytorch

      Public
      Tensors and Dynamic neural networks in Python with strong GPU acceleration
      pytorchrocm
      Python
      •
      Other
      •23k•219•105•23•Updated Oct 30, 2024Oct 30, 2024

    • jax

      Public
      Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
      Python
      •
      Apache License 2.0
      •2.8k•18•0•2•Updated Oct 30, 2024Oct 30, 2024

    • ROCmValidationSuite

      Public
      A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high-performance computing environments
      C++
      •
      MIT License
      •37•65•1•7•Updated Oct 30, 2024Oct 30, 2024

    • rocprofiler-sdk

      Public
      C++
      •
      MIT License
      •7•13•9•4•Updated Oct 30, 2024Oct 30, 2024

    • rocAL

      Public
      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.
      C++
      •
      MIT License
      •14•11•8•1•Updated Oct 30, 2024Oct 30, 2024

    • hipRAND

      Public
      Random number library that generate pseudo-random and quasi-random numbers.
      C++
      •
      MIT License
      •23•23•0•5•Updated Oct 30, 2024Oct 30, 2024

    • MIVisionX

      Public
      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.
      machine-learningcomputer-visionneural-networkopenclinferenceamd-openclvirtual-realityrocminference-engineopenvx
      C++
      •
      MIT License
      •73•184•12•3•Updated Oct 30, 2024Oct 30, 2024

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.3k•135•28•8•Updated Oct 30, 2024Oct 30, 2024

    • vllm

      Public
      A high-throughput and memory-efficient inference and serving engine for LLMs
      Python
      •
      Apache License 2.0
      •4.4k•43•0•20•Updated Oct 30, 2024Oct 30, 2024

    • MITuna

      Public
      Python
      •
      MIT License
      •0•7•0•1•Updated Oct 30, 2024Oct 30, 2024

    • rccl

      Public
      ROCm Communication Collectives Library (RCCL)
      C++
      •
      Other
      •117•263•10•18•Updated Oct 29, 2024Oct 29, 2024

    • aomp

      Public
      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.
      amdllvmopenmpclangfortran-compilerrocm
      Fortran
      •
      Apache License 2.0
      •46•206•7•40•Updated Oct 29, 2024Oct 29, 2024

    • triton

      Public
      Development repository for the Triton language and compiler
      C++
      •
      MIT License
      •1.6k•90•9•42•Updated Oct 29, 2024Oct 29, 2024

    • rocm-spack

      Public
      A flexible package manager that supports multiple versions, configurations, platforms, and compilers.
      Python
      •
      Other
      •2.3k•3•0•18•Updated Oct 29, 2024Oct 29, 2024

    • aotriton

      Public
      Ahead of Time (AOT) Triton Math Library
      Python
      •
      MIT License
      •13•40•9•4•Updated Oct 29, 2024Oct 29, 2024

    • tensorflow-upstream

      Public
      TensorFlow ROCm port
      C++
      •
      Apache License 2.0
      •74k•685•92•60•Updated Oct 29, 2024Oct 29, 2024