2 unstable releases

0.15.0 Oct 28, 2024
0.14.0 Aug 27, 2024

#1066 in Machine learning

Download history 281/week @ 2024-08-24 210/week @ 2024-08-31 255/week @ 2024-09-07 293/week @ 2024-09-14 351/week @ 2024-09-21 307/week @ 2024-09-28 218/week @ 2024-10-05 289/week @ 2024-10-12 220/week @ 2024-10-19 495/week @ 2024-10-26 476/week @ 2024-11-02 318/week @ 2024-11-09 343/week @ 2024-11-16 447/week @ 2024-11-23 605/week @ 2024-11-30

1,797 downloads per month
Used in 15 crates (via burn-core)

MIT/Apache

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

~38–73MB
~1.5M SLoC