candle/candle-kernels
Laurent Mazare 121a71e01f
Fix the silu cuda kernel. (#1710)
2024-02-14 11:08:18 +01:00
..
src Fix the silu cuda kernel. (#1710) 2024-02-14 11:08:18 +01:00
Cargo.toml Bump the crate version to 0.4.0. (#1658) 2024-02-04 19:08:01 +01:00
README.md Revert "Add the layer norm files. (#222)" (#223) 2023-07-22 16:51:11 +01:00
build.rs Moving to a proper build crate `bindgen_cuda`. (#1531) 2024-01-07 12:29:24 +01:00

README.md

candle-kernels

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