huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
fa98ca0c35861e15a532528ed27df0bd40bb4ce5
candle/candle-kernels
History
Laurent Mazare c950a5c6b1 Cuda support for the mnist training. (#277) 
* Cuda support for the mnist training.

* min/max fix + testing.

* Add the argmin/argmax tests.

* More cuda support for argmin/argmax.

* Cuda kernels for argmin and argmax.
2023-07-29 19:48:04 +01:00
..
src
Cuda support for the mnist training. (#277)
2023-07-29 19:48:04 +01:00
build.rs
Revert "Add the layer norm files. (#222)" (#223)
2023-07-22 16:51:11 +01:00
Cargo.toml
Refactor the hierarchy.
2023-06-27 11:57:27 +02: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.