Lib.rs

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

candle-kernels

CUDA kernels for Candle

by Laurent Mazare, Nicolas Patry and 97 contributors

20 releases (7 breaking)

0.8.1 Dec 7, 2024
0.8.0 Nov 12, 2024
0.7.2 Sep 29, 2024
0.6.0 Jun 29, 2024
0.3.1 Nov 11, 2023

#750 in Machine learning

Download history 637/week @ 2024-09-09 1207/week @ 2024-09-16 1311/week @ 2024-09-23 1107/week @ 2024-09-30 754/week @ 2024-10-07 999/week @ 2024-10-14 704/week @ 2024-10-21 576/week @ 2024-10-28 885/week @ 2024-11-04 734/week @ 2024-11-11 874/week @ 2024-11-18 972/week @ 2024-11-25 886/week @ 2024-12-02 1238/week @ 2024-12-09 789/week @ 2024-12-16 548/week @ 2024-12-23

3,549 downloads per month
Used in 46 crates (3 directly)

MIT/Apache

240KB
5.5K SLoC

CUDA 5K SLoC // 0.0% comments Rust 21 SLoC

candle-kernels

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

No runtime deps

~145KB