rocm

There are 12 repositories under rocm topic.

vllm-project / vllm
A high-throughput and memory-efficient inference and serving engine for LLMs
amd cuda deepseek gpt hpu inference inferentia llama llm llm-serving llmops mlops model-serving pytorch qwen rocm tpu trainium transformer xpu
Language:Python 58283
tracel-ai / burn
Burn is a next generation Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
autodiff deep-learning machine-learning rust scientific-computing ndarray tensor neural-network pytorch cross-platform kernel-fusion onnx wasm webgpu cuda metal rocm vulkan
Language:Rust 12907
apache / tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
compiler tensor deep-learning gpu opencl metal performance javascript rocm tvm vulkan spirv machine-learning
Language:Python 12629
cupy / cupy
NumPy & SciPy for GPU
cuda cudnn cublas cusolver nccl python numpy cupy curand cusparse gpu cutensor scipy nvtx nvrtc tensor cusparselt rocm
Language:Python 10498
LMCache / LMCache
Supercharge Your LLM with the Fastest KV Cache Layer
amd cuda fast inference kv-cache llm pytorch rocm speed vllm
Language:Python 5301
gpustack / gpustack
Simple, scalable AI model deployment on GPU clusters
ascend cuda deepseek distributed-inference genai inference llama llamacpp llm maas metal openai qwen rocm vllm mindie llm-inference llm-serving local-ai heterogeneous-cluster
Language:Python 3718
lshqqytiger / stable-diffusion-webui-amdgpu
Stable Diffusion web UI
ai ai-art amd amdgpu deep-learning directml hip image-generation image2image img2img radeon rocm stable-diffusion text2image torch txt2img web
Language:Python 2216
deepmd-kit
deepmodeling / deepmd-kit
A deep learning package for many-body potential energy representation and molecular dynamics
deep-learning molecular-dynamics deepmd lammps potential-energy python tensorflow cpp cuda rocm ipi ase computational-chemistry materials-science c nodejs pytorch jax paddle machine-learning-potential
Language:Python 1752
dmlc / nnvm
computation-graph deep-learning optimization deployment nnvm tvm cuda opencl rocm metal
Language:C++ 1655
aphrodite-engine / aphrodite-engine
Large-scale LLM inference engine
api-rest inference-engine machine-learning cuda inferentia rocm intel lora speculative-decoding tpu
Language:C++ 1548
stdgpu
stotko / stdgpu
stdgpu: Efficient STL-like Data Structures on the GPU
gpu gpu-computing gpu-acceleration gpgpu data-structures stl stl-like stl-containers cpp cpp17 cpp20 modern-cpp cuda openmp rocm hip
Language:C++ 1233
patientx / ComfyUI-Zluda
The most powerful and modular stable diffusion GUI, api and backend with a graph/nodes interface. Now ZLUDA enhanced for better AMD GPU performance.
amd comfyui cuda rocm stable-diffusion windows zluda
Language:Python 597
devnen / Chatterbox-TTS-Server
Self-host the powerful Chatterbox TTS model. This server offers a user-friendly Web UI, flexible API endpoints (incl. OpenAI compatible), predefined voices, voice cloning, and large audiobook-scale text processing. Runs accelerated on NVIDIA (CUDA), AMD (ROCm), and CPU.
ai api-server audio-generation chatterbox cuda fastapi huggingface openai-api python pytorch speech-synthesis speech-synthesis-api text-to-speech tts tts-api voice-cloning web-ui chatterbox-tts rocm
Language:Python 524
ROCm / ROCm-docker
Dockerfiles for the various software layers defined in the ROCm software platform
rocm docker
Language:Shell 490
alpaka-group / alpaka
Abstraction Library for Parallel Kernel Acceleration :llama:
cuda hpc gpu rocm hip openmp openacc heterogeneous-parallel-programming cpp header-only tbb cpp17
Language:C++ 392
ROCm / rocBLAS
[DEPRECATED] Moved to ROCm/rocm-libraries repo
blas rocm hip
Language:C++ 382
QMCPACK / qmcpack
Main repository for QMCPACK, an open-source production level many-body ab initio Quantum Monte Carlo code for computing the electronic structure of atoms, molecules, and solids with full performance portable GPU support
quantum-monte-carlo electronic-structure c-plus-plus high-performance-computing quantum-chemistry cuda gpu hpc mpi rocm oneapi
Language:C++ 356
ROCm / k8s-device-plugin
Kubernetes (k8s) device plugin to enable registration of AMD GPU to a container cluster
kubernetes kubernetes-device-plugins k8s rocm
Language:Go 344
agenium-scale / nsimd
Agenium Scale vectorization library for CPUs and GPUs
simd simd-programming sse2 sse42 avx avx2 avx512 neon neon128 aarch64 sve vectorization-library simd-instructions cuda rocm cpp20 cpp20-library hpc simd-library
Language:C 333
JuliaGPU / AMDGPU.jl
AMD GPU (ROCm) programming in Julia
amdgpu gpu gpu-programming julia rocm
Language:Julia 315
ROCm / pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch rocm
Language:Python 240
ROCm / aomp
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.
amd llvm clang openmp rocm fortran-compiler
Language:Fortran 228
LLNL / hiop
HPC solver for nonlinear optimization problems
hpc nonlinear-optimization nonlinear-programming nonlinear-programming-algorithms interior-point-method parallel-programming mpi bfgs quasi-newton constrained-optimization solver optimization acopf gpu-support cuda math-physics radiuss interior-point-optimizer nonsmooth-optimization rocm
Language:C++ 221
COSMA
eth-cscs / COSMA
Distributed Communication-Optimal Matrix-Matrix Multiplication Algorithm
matrix-multiplication mpi linear-algebra gpu-acceleration cuda communication-optimal rocm pdgemm scalapack matmul
Language:C++ 209
supranational / sppark
Zero-knowledge template library
cuda bls12-377 bls12-381 pasta-curves zero-knowledge zero-knowledge-proofs zk-snarks zk-starks ntt rocm
Language:Cuda 203
MIVisionX
ROCm / MIVisionX
MIVisionX toolkit is a set of comprehensive computer vision and machine intelligence libraries, utilities, and applications bundled into a single toolkit. AMD MIVisionX also delivers a highly optimized open-source implementation of the Khronos OpenVX™ and OpenVX™ Extensions.
computer-vision openvx neural-network opencl rocm inference inference-engine onnx nnef winml windows-machine-learning openvx-neural-network machine-learning ryzen virtual-reality amd-opencl amd-openvx openvx-extensions khronos-openvx amd-opencv
Language:C++ 199
ROCm / rocFFT
[DEPRECATED] Moved to ROCm/rocm-libraries repo
fft rocm hip amd fast fourier gpu transform
Language:C++ 195
ROCm / rocPRIM
[DEPRECATED] Moved to ROCm/rocm-libraries repo
parallel primitive hip rocm cuda amd gpu
Language:C++ 175
virchau13 / automatic1111-webui-nix
AUTOMATIC1111/stable-diffusion-webui for CUDA and ROCm on NixOS
nix nixos stable-diffusion stable-diffusion-webui cuda rocm
Language:Nix 169
ROCm / gpufort
GPUFORT: S2S translation tool for CUDA Fortran and Fortran+X in the spirit of hipify
gpgpu gpu cuda rocm hip fortran cuda-fortran openacc openmp interoperability
Language:Fortran 165
JuliaGPU / AcceleratedKernels.jl
Cross-architecture parallel algorithms for Julia's CPU and GPU backends. Targets multithreaded CPUs, and GPUs via Intel oneAPI, AMD ROCm, Apple Metal, Nvidia CUDA.
amd apple cuda gpu intel julia metal nvidia oneapi parallel rocm standard-library
Language:Julia 159
SIRIUS
electronic-structure / SIRIUS
Domain specific library for electronic structure calculations
density-functional-theory electronic-structure-calculations pseudopotential full-potential planewave lapw gpu cuda rocm mpi
Language:C++ 151
ROCm / hipBLAS
[DEPRECATED] Moved to ROCm/rocm-libraries repo
blas cuda rocm hip
Language:C++ 148
GPUOpen-ProfessionalCompute-Libraries / amdovx-core
AMD OpenVX Core -- a sub-module of amdovx-modules:
openvx radeon-instinct-mi-series radeon-vega-series rocm radeon-open-compute amd-openvx opencl khronos-openvx amdgpu cpu linux vx-loomsl
Language:C++ 147
l1na-forever / stable-diffusion-rocm-docker
Stable Diffusion Docker image preconfigured for usage with AMD Radeon cards
docker rocm stable-diffusion ati radeon amd amdgpu automatic1111 docker-image
Language:Dockerfile 139
Grench6 / RX580-rocM-tensorflow-ubuntu20.4-guide
Install guide of ROCm and Tensorflow on Ubuntu for the RX580
rocm tensorflow-rocm
127