Skip to content

@ROCm AMD ROCm™ Software

Change the repository type filter

All

    Repositories list

    301 repositories

    • triton

      Public
      Development repository for the Triton language and compiler
      Python
      •
      MIT License
      •1.8k•107•8•51•Updated Feb 17, 2025Feb 17, 2025

    • tensorflow-upstream

      Public
      TensorFlow ROCm port
      C++
      •
      Apache License 2.0
      •75k•690•44•76•Updated Feb 17, 2025Feb 17, 2025

    • pytorch

      Public
      Tensors and Dynamic neural networks in Python with strong GPU acceleration
      pytorchrocm
      Python
      •
      Other
      •23k•221•60•47•Updated Feb 17, 2025Feb 17, 2025

    • transformers

      Public
      🤗Transformers: State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0.
      Python
      •
      Apache License 2.0
      •28k•4•0•5•Updated Feb 17, 2025Feb 17, 2025

    • vllm

      Public
      A high-throughput and memory-efficient inference and serving engine for LLMs
      Python
      •
      Apache License 2.0
      •5.7k•65•6•27•Updated Feb 17, 2025Feb 17, 2025

    • xla

      Public
      A machine learning compiler for GPUs, CPUs, and ML accelerators
      C++
      •
      Apache License 2.0
      •503•3•0•19•Updated Feb 17, 2025Feb 17, 2025

    • rocprofiler-compute

      Public
      Advanced Profiling and Analytics for AMD Hardware
      linuxhpcperformance-analysisprofilinghardware-countersgpu-kernels
      Python
      •
      MIT License
      •51•140•52•12•Updated Feb 17, 2025Feb 17, 2025

    • rocMLIR

      Public
      MLIR
      •
      Other
      •39•137•1•21•Updated Feb 17, 2025Feb 17, 2025

    • 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
      •13k•133•18•8•Updated Feb 17, 2025Feb 17, 2025

    • rocm-docs-core

      Public
      ROCm Documentation Python package for ReadTheDocs build standardization
      CSS
      •
      Other
      •16•14•8•4•Updated Feb 17, 2025Feb 17, 2025

    • Megatron-LM

      Public
      Ongoing research training transformer models at scale
      Python
      •
      Other
      •2.6k•15•0•11•Updated Feb 17, 2025Feb 17, 2025

    • jax

      Public
      Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
      Python
      •
      Apache License 2.0
      •2.9k•19•0•15•Updated Feb 17, 2025Feb 17, 2025

    • 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
      •48•211•3•44•Updated Feb 17, 2025Feb 17, 2025

    • AMDMIGraphX

      Public
      AMD's graph optimization engine.
      C++
      •
      MIT License
      •95•208•354•50•Updated Feb 17, 2025Feb 17, 2025

    • rocSPARSE

      Public
      Next generation SPARSE implementation for ROCm platform
      C++
      •
      MIT License
      •56•119•1•0•Updated Feb 17, 2025Feb 17, 2025

    • aiter

      Public
      AI Tensor Engine for ROCm
      Cuda
      •
      MIT License
      •6•23•5•5•Updated Feb 17, 2025Feb 17, 2025

    • composable_kernel

      Public
      Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
      C++
      •
      Other
      •147•346•25•56•Updated Feb 17, 2025Feb 17, 2025

    • HIP

      Public
      HIP: C++ Heterogeneous-Compute Interface for Portability
      hiphip-runtimehipifyhip-kernel-languagehip-portabilitycuda
      C++
      •
      MIT License
      •547•3.9k•20•41•Updated Feb 17, 2025Feb 17, 2025

    • ROCm

      Public
      AMD ROCm™ Software - GitHub Home
      documentation
      Shell
      •
      MIT License
      •405•5k•113•15•Updated Feb 17, 2025Feb 17, 2025

    • apex

      Public
      A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.4k•19•13•6•Updated Feb 17, 2025Feb 17, 2025

    • 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
      •105•79•9•74•Updated Feb 17, 2025Feb 17, 2025

    • MIOpen

      Public
      AMD's Machine Intelligence Library
      Assembly
      •
      Other
      •242•1.1k•248•64•Updated Feb 17, 2025Feb 17, 2025

    • aotriton

      Public
      Ahead of Time (AOT) Triton Math Library
      Python
      •
      MIT License
      •18•52•12•1•Updated Feb 17, 2025Feb 17, 2025

    • clr

      Public
      C++
      •
      MIT License
      •53•117•16•21•Updated Feb 17, 2025Feb 17, 2025

    • ROCgdb

      Public
      This is ROCgdb, the ROCm source-level debugger for Linux, based on GDB, the GNU source-level debugger.
      C
      •
      GNU General Public License v2.0
      •10•53•5•1•Updated Feb 17, 2025Feb 17, 2025

    • xformers

      Public
      Hackable and optimized Transformers building blocks, supporting a composable construction.
      Python
      •
      Other
      •648•24•9•2•Updated Feb 17, 2025Feb 17, 2025

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.5k•154•20•6•Updated Feb 17, 2025Feb 17, 2025

    • rocm-examples

      Public
      A collection of examples for the ROCm software stack
      C++
      •
      MIT License
      •48•185•2•1•Updated Feb 17, 2025Feb 17, 2025

    • 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
      •40•67•0•10•Updated Feb 17, 2025Feb 17, 2025

    • rccl

      Public
      ROCm Communication Collectives Library (RCCL)
      C++
      •
      Other
      •136•297•14•15•Updated Feb 17, 2025Feb 17, 2025