huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ad8a62dbf50b0771e392662cf444917a72aed844
candle/candle-kernels
History
Laurent Mazare ad8a62dbf5 Add tanh. (#675) 
* Add tanh.

* Use tanh in the lstm block.

* Add a test for tanh forward and backward passes.
2023-08-30 13:54:50 +01:00
..
src
Add tanh. (#675)
2023-08-30 13:54:50 +01:00
build.rs
Revert "Add the layer norm files. (#222)" (#223)
2023-07-22 16:51:11 +01:00
Cargo.toml
Add some documentation. (#673)
2023-08-30 11:54:00 +01:00
README.md
Revert "Add the layer norm files. (#222)" (#223)
2023-07-22 16:51:11 +01:00

README.md

candle-kernels

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