Lib.rs

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

no-std burn-cuda

CUDA backend for the Burn framework

by Nathaniel Simard

1 unstable release

0.14.0 Aug 27, 2024

#1022 in Machine learning

Download history 352/week @ 2024-08-26 183/week @ 2024-09-02 304/week @ 2024-09-09

839 downloads per month
Used in 7 crates (via burn-core)

MIT/Apache

1.5MB
37K SLoC

Burn-Cuda

This backend is still a work in progress and not ready to be used.

See #1525

Dependencies

~34–68MB
~1.5M SLoC