candle/candle-kernels
Laurent Mazare 8097559c1a
Move the candle version to 0.7.1. (#2495)
2024-09-22 20:44:39 +02:00
..
src Add the layernorm specialized op. (#2212) 2024-05-24 15:58:01 +02:00
Cargo.toml Move the candle version to 0.7.1. (#2495) 2024-09-22 20:44:39 +02:00
README.md Revert "Add the layer norm files. (#222)" (#223) 2023-07-22 16:51:11 +01:00
build.rs Ensure that the kernels get rebuilt on cuh changes. (#1954) 2024-03-28 06:56:48 +01:00

README.md

candle-kernels

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