Thanks to visit codestin.com
Credit goes to lib.rs

Lib.rs

› Science › Machine learning › burn-common
#deep-learning #rocm #hip #machine-learning #gpu

no-std burn-rocm

ROCm HIP backend for the Burn framework

by Nathaniel Simard and 96 contributors

13 unstable releases (3 breaking)

Uses new Rust 2024

new 0.20.1 Jan 23, 2026
0.20.0-pre.6 Dec 18, 2025
0.20.0-pre.3 Nov 24, 2025
0.18.0 Jul 18, 2025

#2285 in Machine learning

Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App Codestin Search App

43,029 downloads per month
Used in 99 crates (4 directly)

MIT/Apache

1.5MB
32K SLoC

burn-rocm

Backend using ROCm HIP runtime.

To execute the tests for this backend set an environment variable called ROCM_PATH or CUBECL_ROCM_PATH to the installation path of ROCm. It is often /opt/rocm.

For now this backend requires the version 6.2.2 of ROCm or a compatible version.

Dependencies

~68–110MB
~2M SLoC