Lib.rs

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

candle-kernels

CUDA kernels for Candle

by Laurent Mazare, Nicolas Patry and 97 contributors

21 releases (7 breaking)

new 0.8.2 Jan 7, 2025
0.8.1 Dec 7, 2024
0.8.0 Nov 12, 2024
0.6.0 Jun 29, 2024
0.3.1 Nov 11, 2023

#756 in Machine learning

Download history 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 532/week @ 2024-12-30 1009/week @ 2025-01-06

3,009 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