5 releases (breaking)

Uses new Rust 2024

0.5.0 Apr 23, 2025
0.4.0 Jan 14, 2025
0.3.0 Oct 28, 2024
0.2.0 Aug 27, 2024
0.1.1 Jul 19, 2024

#1310 in Machine learning

Download history 1151/week @ 2025-01-17 1280/week @ 2025-01-24 1501/week @ 2025-01-31 1193/week @ 2025-02-07 1241/week @ 2025-02-14 5097/week @ 2025-02-21 3250/week @ 2025-02-28 3811/week @ 2025-03-07 2967/week @ 2025-03-14 2720/week @ 2025-03-21 2392/week @ 2025-03-28 2013/week @ 2025-04-04 1990/week @ 2025-04-11 5041/week @ 2025-04-18 6004/week @ 2025-04-25 3345/week @ 2025-05-02

16,775 downloads per month
Used in 32 crates (via cubecl)

MIT/Apache

1MB
30K SLoC

Cuda runtime

The runtime uses the lower level primitives from cudarc to compile generated CUDA code into a ptx and execute it at runtime.

Setup

By default, this runtime uses the latest CUDA version. If the default features are not enabled, the CUDA version detected during build time will be used instead. To specify a fixed CUDA version, disable the default features and add cudarc to your dependencies with the appropriate feature flag:

cudarc = { version = "same as burn", features = ["cuda-11040"] }

Dependencies

~12–27MB
~389K SLoC