4 releases (2 breaking)

new 0.16.1 Apr 3, 2025
0.16.0 Jan 14, 2025
0.15.0 Oct 28, 2024
0.14.0 Aug 27, 2024

#1155 in Machine learning

Download history 636/week @ 2024-12-16 284/week @ 2024-12-23 284/week @ 2024-12-30 597/week @ 2025-01-06 754/week @ 2025-01-13 938/week @ 2025-01-20 1304/week @ 2025-01-27 1173/week @ 2025-02-03 1092/week @ 2025-02-10 2835/week @ 2025-02-17 4198/week @ 2025-02-24 3002/week @ 2025-03-03 3508/week @ 2025-03-10 2610/week @ 2025-03-17 2680/week @ 2025-03-24 2270/week @ 2025-03-31

11,502 downloads per month
Used in 19 crates (via burn-core)

MIT/Apache

2MB
46K SLoC

Burn CUDA Backend

Burn CUDA backend

Current Crates.io Version license

This crate provides a CUDA backend for Burn using the cubecl and cudarc crates.

Usage Example

#[cfg(feature = "cuda")]
mod cuda {
    use burn_autodiff::Autodiff;
    use burn_cuda::{Cuda, CudaDevice};
    use mnist::training;

    pub fn run() {
        let device = CudaDevice::default();
        training::run::<Autodiff<Cuda<f32, i32>>>(device);
    }
}

Dependencies

Requires CUDA 12.x to be installed and on the PATH.

Dependencies

~44–78MB
~1.5M SLoC