Skip to content

@ROCm AMD ROCm™ Software

Change the repository type filter

All

    Repositories list

    290 repositories

    • HIP

      Public
      HIP: C++ Heterogeneous-Compute Interface for Portability
      hiphip-runtimehipifyhip-kernel-languagehip-portabilitycuda
      C++
      •
      MIT License
      •539•3.8k•28•43•Updated Nov 22, 2024Nov 22, 2024

    • composable_kernel

      Public
      Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
      C++
      •
      Other
      •129•317•30•46•Updated Nov 22, 2024Nov 22, 2024

    • MIOpen

      Public
      AMD's Machine Intelligence Library
      Assembly
      •
      Other
      •230•1.1k•250•54•Updated Nov 22, 2024Nov 22, 2024

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.3k•140•24•8•Updated Nov 22, 2024Nov 22, 2024

    • jax

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

    • ROCm

      Public
      AMD ROCm™ Software - GitHub Home
      documentation
      Shell
      •
      MIT License
      •389•4.7k•111•14•Updated Nov 22, 2024Nov 22, 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
      •89•63•8•72•Updated Nov 22, 2024Nov 22, 2024

    • triton

      Public
      Development repository for the Triton language and compiler
      C++
      •
      MIT License
      •1.7k•96•10•43•Updated Nov 22, 2024Nov 22, 2024

    • AMDMIGraphX

      Public
      AMD's graph optimization engine.
      C++
      •
      MIT License
      •86•187•344•40•Updated Nov 22, 2024Nov 22, 2024

    • rocMLIR

      Public
      MLIR
      •40•129•1•16•Updated Nov 22, 2024Nov 22, 2024

    • torch_migraphx

      Public
      Libraries integrating migraphx with pytorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1•5•13•4•Updated Nov 22, 2024Nov 22, 2024

    • ROCR-Runtime

      Public
      ROCm Platform Runtime: ROCr a HPC market enhanced HSA based runtime
      C++
      •
      Other
      •110•224•12•29•Updated Nov 22, 2024Nov 22, 2024

    • rocprofiler-systems

      Public
      ROCm Systems Profiler
      C++
      •
      MIT License
      •3•7•1•5•Updated Nov 22, 2024Nov 22, 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•123•30•13•Updated Nov 22, 2024Nov 22, 2024

    • pytorch

      Public
      Tensors and Dynamic neural networks in Python with strong GPU acceleration
      pytorchrocm
      Python
      •
      Other
      •23k•219•103•37•Updated Nov 22, 2024Nov 22, 2024

    • rocm-examples

      Public
      A collection of examples for the ROCm software stack
      C++
      •
      MIT License
      •41•168•6•3•Updated Nov 22, 2024Nov 22, 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•66•0•11•Updated Nov 22, 2024Nov 22, 2024

    • FBGEMM

      Public
      FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
      C++
      •
      Other
      •504•1•0•10•Updated Nov 22, 2024Nov 22, 2024

    • tensorflow-upstream

      Public
      TensorFlow ROCm port
      C++
      •
      Apache License 2.0
      •74k•688•90•60•Updated Nov 22, 2024Nov 22, 2024

    • xla

      Public
      A machine learning compiler for GPUs, CPUs, and ML accelerators
      C++
      •
      Apache License 2.0
      •436•2•0•14•Updated Nov 22, 2024Nov 22, 2024

    • Tensile

      Public
      Stretching GPU performance for GEMMs and tensor contractions.
      pythonmachine-learningamdgpuassemblyopencldnnmatrix-multiplicationneural-networksgpu-acceleration
      Python
      •
      MIT License
      •151•223•8•6•Updated Nov 22, 2024Nov 22, 2024

    • rocprofiler

      Public
      ROC profiler library. Profiling with perf-counters and derived metrics.
      C
      •
      MIT License
      •49•132•9•21•Updated Nov 22, 2024Nov 22, 2024

    • clr

      Public
      C++
      •
      MIT License
      •50•104•12•15•Updated Nov 22, 2024Nov 22, 2024

    • 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•52•5•1•Updated Nov 22, 2024Nov 22, 2024

    • hip-tests

      Public
      C++
      •
      MIT License
      •31•30•1•23•Updated Nov 22, 2024Nov 22, 2024

    • rocprofiler-sdk

      Public
      C++
      •
      MIT License
      •9•16•10•5•Updated Nov 22, 2024Nov 22, 2024

    • aotriton

      Public
      Ahead of Time (AOT) Triton Math Library
      Python
      •
      MIT License
      •15•41•9•2•Updated Nov 22, 2024Nov 22, 2024

    • rocm-spack

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

    • Megatron-LM

      Public
      Ongoing research training transformer models at scale
      Python
      •
      Other
      •2.4k•9•0•3•Updated Nov 22, 2024Nov 22, 2024

    • rocSOLVER

      Public
      Next generation LAPACK implementation for ROCm platform
      linear-algebralapackrocm
      C++
      •
      Other
      •53•95•0•16•Updated Nov 22, 2024Nov 22, 2024