Lib.rs

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

candle-kernels

CUDA kernels for Candle

by Laurent Mazare, Nicolas Patry and 97 contributors

13 releases (4 breaking)

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

#476 in Machine learning

Download history 1055/week @ 2024-01-26 700/week @ 2024-02-02 832/week @ 2024-02-09 1039/week @ 2024-02-16 1132/week @ 2024-02-23 788/week @ 2024-03-01 895/week @ 2024-03-08 916/week @ 2024-03-15 1418/week @ 2024-03-22 1153/week @ 2024-03-29 858/week @ 2024-04-05 984/week @ 2024-04-12 660/week @ 2024-04-19 508/week @ 2024-04-26 736/week @ 2024-05-03 456/week @ 2024-05-10

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

MIT/Apache

235KB
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