GPU Computing

GPUs are widely used in machine learning tasks including both training and inference, making them key components a performant machine learning ecosystem. Rust's GPU support, particularly around Geneneral-Purpose GPU (GPGPU) and compute shaders is incomplete, but a number of efforts including the WebGPU ecosystem, as well as CUDA and OpenCL are making progress.

wgpu [ crate · repo · docs ]

Rusty WebGPU API wrapper

Last Commit: 2022-12-04 Last Published: 2022-11-30

vulkano [ crate · repo · docs ]

Safe wrapper for the Vulkan graphics API

Last Commit: 2022-12-02 Last Published: 2022-11-09

ocl [ crate · repo · docs ]

OpenCL bindings and interfaces for Rust.

Last Commit: 2022-09-16 Last Published: 2022-08-23

arrayfire [ crate · repo · docs ]

ArrayFire is a high performance software library for parallel computing with an easy-to-use API. Its array based function set makes parallel programming simple. ArrayFire's multiple backends (CUDA, OpenCL and native CPU) make it platform independent and highly portable. A few lines of code in ArrayFire can replace dozens of lines of parallel computing code, saving you valuable time and lowering development costs. This crate provides Rust bindings for ArrayFire library.

Last Commit: 2022-10-14 Last Published: 2021-01-10

rustacuda [ crate · repo · docs ]

CUDA Driver API Wrapper

Last Commit: 2022-05-04 Last Published: 2021-07-28

blas [ crate · repo · docs ]

The package provides wrappers for BLAS (Fortran).

Last Commit: 2022-02-23 Last Published: 2021-04-10

cuda_std [ crate · repo · docs ]

Standard library for CUDA with rustc_codegen_nvvm

Last Commit: 2022-11-20 Last Published: 2022-02-07

wonnx [ crate · repo · docs ]

Wonnx is an ONNX runtime based on wgpu aimed at being a universal GPU runtime, written in Rust.

Last Commit: 2022-11-28 Last Published: 2022-07-31

emu_core [ crate · repo · docs ]

A low-level modular, extensible GPGPU compute library with cross-platform support

Last Commit: 2022-06-17 Last Published: 2020-04-09

collenchyma [ crate · repo · docs ]

high-performance computation on any hardware

Last Commit: 2016-04-23 Last Published: 2016-02-21

coaster [ crate · repo · docs ]

high-performance computation on any hardware

Last Commit: 2019-12-28 Last Published: 2021-11-21

cudnn [ crate · repo · docs ]

safe Rust wrapper for CUDA's cuDNN

Last Commit: 2021-07-21 Last Published: 2016-03-02

accel [ crate · repo · docs ]

GPGPU Framework for Rust

Last Published: 2020-05-24

cublas [ crate · repo · docs ]

safe Rust wrapper for CUDA's cuBLAS

Last Commit: 2018-01-14 Last Published: 2016-02-04

em [ crate · repo · docs ]

A macro for executing a subset of Rust with OpenCL

Last Published: 2019-10-21

cuda [ crate · repo · docs ]

CUDA bindings.

Last Commit: 2019-12-27 Last Published: 2019-06-18

cumath [ crate · repo · docs ]

Cuda-based matrix/vector computations

Last Commit: 2020-02-25 Last Published: 2018-05-23

parenchyma [ crate · repo · docs ]

A high-performance computing (HPC) framework

Last Commit: 2018-08-23 Last Published: 2017-03-18

Do you know of a crate that is missing? Create an issue or a PR.
Looking for something you didn't find? Try asking on Zulip chat.