Lib.rs

› Science › Machine learning
#machine-learning #tensor #blas

candle-kernels

CUDA kernels for Candle

by Laurent Mazare, Nicolas Patry and 85 contributors

12 unstable releases (3 breaking)

0.4.1 Feb 28, 2024
0.3.3 Jan 31, 2024
0.3.2 Dec 20, 2023
0.3.1 Nov 11, 2023

#769 in Machine learning

Download history 619/week @ 2024-01-08 602/week @ 2024-01-15 632/week @ 2024-01-22 1199/week @ 2024-01-29 706/week @ 2024-02-05 830/week @ 2024-02-12 944/week @ 2024-02-19 1237/week @ 2024-02-26 754/week @ 2024-03-04 862/week @ 2024-03-11 967/week @ 2024-03-18 1398/week @ 2024-03-25 1203/week @ 2024-04-01 887/week @ 2024-04-08 876/week @ 2024-04-15 611/week @ 2024-04-22

3,651 downloads per month
Used in 35 crates (2 directly)

MIT/Apache

125KB
3K SLoC

CUDA 3K SLoC // 0.0% comments Rust 17 SLoC

candle-kernels

This crate contains CUDA kernels used from candle. Some of these implementations come from the dfdx crate.

No runtime deps

~145KB