Lib.rs

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

candle-kernels

CUDA kernels for Candle

by Laurent Mazare, Nicolas Patry and 97 contributors

19 releases (7 breaking)

0.8.0 Nov 12, 2024
0.6.0 Jun 29, 2024
0.4.1 Feb 28, 2024
0.3.2 Dec 20, 2023
0.3.1 Nov 11, 2023

#762 in Machine learning

Download history 1027/week @ 2024-07-31 948/week @ 2024-08-07 829/week @ 2024-08-14 974/week @ 2024-08-21 869/week @ 2024-08-28 677/week @ 2024-09-04 676/week @ 2024-09-11 1240/week @ 2024-09-18 1360/week @ 2024-09-25 979/week @ 2024-10-02 806/week @ 2024-10-09 991/week @ 2024-10-16 635/week @ 2024-10-23 721/week @ 2024-10-30 802/week @ 2024-11-06 598/week @ 2024-11-13

2,873 downloads per month
Used in 45 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