Files
candle/candle-examples
Laurent Mazare 1c9e5394a5 Add a custom softmax implementation. (#744)
* Add a custom softmax implementation.

* Add softmaxlastdim to the benchmarks.

* And add a test.

* Support more dtypes.

* Polish the code.

* Use the slow implementation on cuda.

* Add a todo for the cuda kernel.
2023-09-05 14:20:23 +01:00
..
2023-08-29 10:19:44 +02:00
2023-07-26 07:48:10 +01:00
2023-08-30 11:54:00 +01:00

candle-examples