Skip to content

@ROCm AMD ROCm™ Software

Change the repository type filter

All

    Repositories list

    285 repositories

    • AMDMIGraphX

      Public
      AMD's graph optimization engine.
      C++
      •
      MIT License
      •82•184•364•33•Updated Oct 5, 2024Oct 5, 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•15•51•Updated Oct 4, 2024Oct 4, 2024

    • pytorch

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

    • tensorflow-upstream

      Public
      TensorFlow ROCm port
      C++
      •
      Apache License 2.0
      •74k•684•91•53•Updated Oct 4, 2024Oct 4, 2024

    • triton

      Public
      Development repository for the Triton language and compiler
      C++
      •
      MIT License
      •1.6k•89•7•43•Updated Oct 4, 2024Oct 4, 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•203•8•40•Updated Oct 4, 2024Oct 4, 2024

    • flang

      Public
      Mirror of flang repo: The source repo is https://github.com/flang-compiler/flang . Once a day the master branch is updated from the upstream source repo and then locked. AOMP or ROCm developers may commit or create PRs on branch aomp-dev.
      C++
      •
      Other
      •7•5•0•10•Updated Oct 4, 2024Oct 4, 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•2•Updated Oct 4, 2024Oct 4, 2024

    • TransformerEngine

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

    • vllm

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

    • ROCm

      Public
      AMD ROCm™ Software - GitHub Home
      documentation
      Shell
      •
      MIT License
      •374•4.5k•136•13•Updated Oct 4, 2024Oct 4, 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 4, 2024Oct 4, 2024

    • k8s-device-plugin

      Public
      Kubernetes (k8s) device plugin to enable registration of AMD GPU to a container cluster
      kubernetes-device-pluginskubernetesk8srocm
      Go
      •
      Apache License 2.0
      •47•266•12•2•Updated Oct 4, 2024Oct 4, 2024

    • rocm-spack

      Public
      A flexible package manager that supports multiple versions, configurations, platforms, and compilers.
      Python
      •
      Other
      •2.2k•3•0•17•Updated Oct 4, 2024Oct 4, 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•5•6•2•Updated Oct 4, 2024Oct 4, 2024

    • Megatron-LM

      Public
      Ongoing research training transformer models at scale
      Python
      •
      Other
      •2.3k•4•0•1•Updated Oct 4, 2024Oct 4, 2024

    • aotriton

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

    • composable_kernel

      Public
      Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
      C++
      •
      Other
      •115•297•42•33•Updated Oct 4, 2024Oct 4, 2024

    • rocMLIR

      Public
      C++
      •40•123•2•14•Updated Oct 4, 2024Oct 4, 2024

    • rocprofiler-systems

      Public
      ROCm Systems Profiler
      C++
      •
      MIT License
      •2•3•0•3•Updated Oct 4, 2024Oct 4, 2024

    • omnitrace

      Public
      Omnitrace: Application Profiling, Tracing, and Analysis
      pythonlinuxprofilertracingperformance-metricsperformance-analysisprofilingperformance-monitoringcode-coveragehardware-counters
      C++
      •
      MIT License
      •25•293•25•5•Updated Oct 4, 2024Oct 4, 2024

    • HIP

      Public
      HIP: C++ Heterogeneous-Compute Interface for Portability
      hiphip-runtimehipifyhip-kernel-languagehip-portabilitycuda
      C++
      •
      MIT License
      •528•3.7k•56•48•Updated Oct 4, 2024Oct 4, 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•26•10•Updated Oct 4, 2024Oct 4, 2024

    • MIOpen

      Public
      AMD's Machine Intelligence Library
      Assembly
      •
      Other
      •221•1.1k•264•49•Updated Oct 4, 2024Oct 4, 2024

    • rocPRIM

      Public
      ROCm Parallel Primitives
      amdgpuparallelcudaprimitivehiprocm
      C++
      •
      MIT License
      •68•154•1•7•Updated Oct 4, 2024Oct 4, 2024

    • rocSPARSE

      Public
      Next generation SPARSE implementation for ROCm platform
      C++
      •
      MIT License
      •55•115•2•0•Updated Oct 4, 2024Oct 4, 2024

    • rocm-docs-core

      Public
      ROCm Documentation Python package for ReadTheDocs build standardization
      CSS
      •
      Other
      •15•12•11•3•Updated Oct 4, 2024Oct 4, 2024

    • rccl

      Public
      ROCm Communication Collectives Library (RCCL)
      C++
      •
      Other
      •113•254•10•18•Updated Oct 4, 2024Oct 4, 2024

    • rocALUTION

      Public
      Next generation library for iterative sparse solvers for ROCm platform
      cplusplusfortranopenclsolveropenmpmpicudasparse
      C++
      •
      MIT License
      •39•73•4•0•Updated Oct 4, 2024Oct 4, 2024

    • Gromacs

      Public
      ROCm's implementation of Gromacs
      C++
      •
      GNU Lesser General Public License v2.1
      •2•6•4•7•Updated Oct 4, 2024Oct 4, 2024