#machine-learning #blas #tensor

candle-kernels

Candle 的 CUDA 内核

15 个版本 (5 个重大更新)

0.6.0 2024年6月29日
0.4.1 2024年2月28日
0.3.2 2023年12月20日
0.3.1 2023年11月11日

机器学习 中排名 #795

Download history 623/week @ 2024-04-21 668/week @ 2024-04-28 563/week @ 2024-05-05 678/week @ 2024-05-12 1045/week @ 2024-05-19 759/week @ 2024-05-26 727/week @ 2024-06-02 742/week @ 2024-06-09 802/week @ 2024-06-16 1220/week @ 2024-06-23 1242/week @ 2024-06-30 905/week @ 2024-07-07 1290/week @ 2024-07-14 791/week @ 2024-07-21 1141/week @ 2024-07-28 850/week @ 2024-08-04

每月下载量 4,167
39crate中使用(直接使用2个)

MIT/Apache

240KB
5.5K SLoC

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

candle-kernels

此crate包含从candle使用的CUDA内核。其中一些实现来自dfdx crate

无运行时依赖

~145KB