Files
candle/candle-kernels
Laurent Mazare c18a856e76 Add the rounding operators. (#1030)
* Add the rounding operators.

* Avoid tracking gradients for the rounding operations.

* Add some rounding tests.
2023-10-04 17:58:44 +01:00
..
2023-10-04 17:58:44 +01:00
2023-10-01 13:51:57 +01:00

candle-kernels

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