Lib.rs

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

no-std burn-hip

ROCm HIP backend for the Burn framework

by Nathaniel Simard

1 unstable release

0.15.0 Oct 28, 2024

#1165 in Machine learning

Download history 392/week @ 2024-10-28 349/week @ 2024-11-04 273/week @ 2024-11-11 290/week @ 2024-11-18 381/week @ 2024-11-25 494/week @ 2024-12-02 585/week @ 2024-12-09

1,787 downloads per month
Used in 8 crates (via burn-core)

MIT/Apache

2MB
45K 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

~43–78MB
~1.5M SLoC