Skip to content

@ROCm AMD ROCm™ Software

Change the repository type filter

All

    Repositories list

    282 repositories

    • TransformerEngine

      Public
      Python
      •
      Other
      •0•4•2•3•Updated Oct 1, 2024Oct 1, 2024

    • Tensile

      Public
      Stretching GPU performance for GEMMs and tensor contractions.
      pythonmachine-learningamdgpuassemblyopencldnnmatrix-multiplicationneural-networksgpu-acceleration
      Python
      •
      MIT License
      •147•214•15•6•Updated Oct 1, 2024Oct 1, 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•113•25•11•Updated Oct 1, 2024Oct 1, 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
      •80•49•14•48•Updated Oct 1, 2024Oct 1, 2024

    • triton

      Public
      Development repository for the Triton language and compiler
      C++
      •
      MIT License
      •1.6k•89•7•44•Updated Oct 1, 2024Oct 1, 2024

    • vllm

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

    • tensorflow-upstream

      Public
      TensorFlow ROCm port
      C++
      •
      Apache License 2.0
      •74k•684•90•54•Updated Oct 1, 2024Oct 1, 2024

    • composable_kernel

      Public
      Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
      C++
      •
      Other
      •113•297•45•29•Updated Oct 1, 2024Oct 1, 2024

    • AMDMIGraphX

      Public
      AMD's graph optimization engine.
      C++
      •
      MIT License
      •84•184•374•31•Updated Oct 1, 2024Oct 1, 2024

    • rocm_smi_lib

      Public
      ROCm SMI LIB
      sysadmintrace
      C++
      •
      MIT License
      •49•116•18•27•Updated Oct 1, 2024Oct 1, 2024

    • flash-attention

      Public
      Fast and memory-efficient exact attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.2k•131•27•6•Updated Oct 1, 2024Oct 1, 2024

    • rocBLAS

      Public
      Next generation BLAS implementation for ROCm platform
      blashiprocm
      C++
      •
      Other
      •161•339•6•3•Updated Oct 1, 2024Oct 1, 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
      •44•205•8•39•Updated Sep 30, 2024Sep 30, 2024

    • MIOpen

      Public
      AMD's Machine Intelligence Library
      Assembly
      •
      Other
      •221•1.1k•270•52•Updated Sep 30, 2024Sep 30, 2024

    • rocPyDecode

      Public
      rocPyDecode is a set of Python bindings to rocDecode C++ library which provides full HW acceleration for video decoding on AMD GPUs.
      C++
      •
      MIT License
      •5•0•2•3•Updated Sep 30, 2024Sep 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•17•0•1•Updated Sep 30, 2024Sep 30, 2024

    • HIPIFY

      Public
      HIPIFY: Convert CUDA to Portable C++ Code
      hipifyhipify-clanghipify-perlcuda2hip
      C++
      •
      MIT License
      •70•503•21•1•Updated Sep 30, 2024Sep 30, 2024

    • pytorch

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

    • aotriton

      Public
      Ahead of Time (AOT) Triton Math Library
      Python
      •
      MIT License
      •13•36•9•2•Updated Sep 30, 2024Sep 30, 2024

    • rocm-install-on-linux

      Public
      C
      •
      MIT License
      •9•10•0•9•Updated Sep 30, 2024Sep 30, 2024

    • 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•18•13•6•Updated Sep 30, 2024Sep 30, 2024

    • rocMLIR

      Public
      C++
      •40•123•3•12•Updated Sep 30, 2024Sep 30, 2024

    • DeepSpeed

      Public
      DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
      Python
      •
      Apache License 2.0
      •4.1k•4•6•2•Updated Sep 30, 2024Sep 30, 2024

    • ROCm

      Public
      AMD ROCm™ Software - GitHub Home
      documentation
      Shell
      •
      MIT License
      •373•4.5k•142•15•Updated Sep 30, 2024Sep 30, 2024

    • HIP

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

    • xgboost

      Public
      Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow
      C++
      •
      Apache License 2.0
      •8.7k•1•2•0•Updated Sep 30, 2024Sep 30, 2024

    • xla

      Public
      A machine learning compiler for GPUs, CPUs, and ML accelerators
      C++
      •
      Apache License 2.0
      •407•2•0•11•Updated Sep 30, 2024Sep 30, 2024

    • rocm-blogs

      Public
      Jupyter Notebook
      •7•28•0•1•Updated Sep 30, 2024Sep 30, 2024

    • hipTensor

      Public
      AMD’s C++ library for accelerating tensor primitives
      C++
      •
      MIT License
      •16•34•0•4•Updated Sep 30, 2024Sep 30, 2024

    • tensorcast

      Public
      Python
      •
      MIT License
      •2•3•0•0•Updated Sep 30, 2024Sep 30, 2024