dnrops
/
candle
mirror of https://github.com/huggingface/candle.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
candle/candle-kernels
History
Laurent Mazare 94c6a8d3d3
Add a dedicated cuda kernel for softmax. (#746) 		2023-09-05 17:53:20 +02:00
..
src Add a dedicated cuda kernel for softmax. (#746) 2023-09-05 17:53:20 +02: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
build.rs 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.