Top 13 rocm open source projects

No description or website provided.

✭ 1,639

C++python CMake shell deep-learning cuda optimization deployment opencl metal computation-graph rocm nnvm tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

✭ 7,494

python C++rust c CMake shell deep-learning machine-learning performance compiler gpu vulkan opencl metal tensor spirv rocm tvm

Numba

NumPy aware dynamic Python compiler using LLVM

✭ 7,090

python c C++shell Batchfile HTML compiler cuda numpy llvm parallel rocm

Cupy

NumPy & SciPy for GPU

✭ 5,625

python cython C++c Cuda Dockerfile gpu cuda numpy tensor scipy cupy cudnn cublas rocm cusolver nccl curand cusparse nvrtc cutensor nvtx cusparselt

ROCArrays.jl

Parallel on the ROCks

✭ 17

julia gpu arrays amdgpu rocm

Hetero-Mark

A Benchmark Suite for Heterogeneous System Computation

✭ 41

Jupyter Notebook C++Cuda CMake c python benchmark gpu opencl cuda hip rocm hetero-mark

rocPRIM

ROCm Parallel Primitives

✭ 95

C++CMake python amd gpu parallel cuda primitive hip rocm

rocRAND

RAND library for HIP programming language

✭ 68

c C++fortran CMake python groovy gpu random cuda rng hip rocm

RET

ROCm Machine Learning and HPC Stack installer

✭ 28

shell Dockerfile deep-learning hpc amd tensorflow pytorch rocm

SIRIUS

Domain specific library for electronic structure calculations

✭ 87

C++python fortran CMake Cuda c gpu mpi cuda density-functional-theory rocm electronic-structure-calculations pseudopotential planewave full-potential lapw

amdovx-modules

AMD OpenVX modules: such as, neural network inference, 360 video stitching, etc.

✭ 106

C++python c video-stitching rocm radeon-open-compute openvx onnx neural-network-inference radeon-instinct-mi-series radeon-vega-series

realcaffe2

The repo is obsolete. Use at your own risk.

✭ 12

C++python Cuda c Objective-C++CMake hpc amd amdgpu rocm caffe2

k8s-device-plugin

Kubernetes (k8s) device plugin to enable registration of AMD GPU to a container cluster

✭ 163

go Dockerfile Smarty kubernetes k8s rocm kubernetes-device-plugins

1-13 of 13 rocm projects