Lib.rs

› Science › Machine learning › Burn
#hip #deep-learning #rocm #machine-learning #gpu #back-end

no-std burn-hip

ROCm HIP backend for the Burn framework

by Nathaniel Simard

2 unstable releases

0.16.0 Jan 14, 2025
0.15.0 Oct 28, 2024

#1121 in Machine learning

Download history 365/week @ 2024-10-27 343/week @ 2024-11-03 269/week @ 2024-11-10 283/week @ 2024-11-17 380/week @ 2024-11-24 513/week @ 2024-12-01 562/week @ 2024-12-08 556/week @ 2024-12-15 223/week @ 2024-12-22 206/week @ 2024-12-29 537/week @ 2025-01-05 679/week @ 2025-01-12 889/week @ 2025-01-19 1178/week @ 2025-01-26 1091/week @ 2025-02-02 857/week @ 2025-02-09

4,062 downloads per month
Used in 10 crates (via burn-core)

MIT/Apache

2MB
46K SLoC

Burn-hip

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

~48–82MB
~1.5M SLoC