candle/candle-kernels
Laurent Mazare 89f53b9d7b
Bump the version number to 0.5.1. (#2155)
* Bump the version number to 0.5.1.

* Fix clippy lints for 1.78.

* More clippy fixes.
2024-05-03 11:17:05 +02:00
..
src Fix sigmoid gradient calculation and move sigmoid into a specialized op (#2114) 2024-04-29 11:04:43 +02:00
Cargo.toml Bump the version number to 0.5.1. (#2155) 2024-05-03 11:17:05 +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.