Lib.rs

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

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

#1263 in Machine learning

Download history 643/week @ 2024-12-10 462/week @ 2024-12-17 205/week @ 2024-12-24 294/week @ 2024-12-31 533/week @ 2025-01-07 880/week @ 2025-01-14 791/week @ 2025-01-21 1255/week @ 2025-01-28 1076/week @ 2025-02-04 872/week @ 2025-02-11 3532/week @ 2025-02-18 3716/week @ 2025-02-25 3057/week @ 2025-03-04 3119/week @ 2025-03-11 2489/week @ 2025-03-18 2318/week @ 2025-03-25

11,490 downloads per month
Used in 12 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

~44–74MB
~1.5M SLoC