candle/candle-kernels
Laurent Mazare 4fe8a02f88
Update the repo location. (#305)
2023-08-02 11:12:18 +01:00
..
src Remove the embedding ops in favor of index-select. (#299) 2023-08-02 05:42:11 +01:00
Cargo.toml Update the repo location. (#305) 2023-08-02 11:12:18 +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.