huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1c9e5394a5056aadc948f9330ea31fea4972e65e
candle/candle-nn
History
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
..
examples
Add a custom softmax implementation. (#744)
2023-09-05 14:20:23 +01:00
src
Add a custom softmax implementation. (#744)
2023-09-05 14:20:23 +01:00
tests
Add a custom softmax implementation. (#744)
2023-09-05 14:20:23 +01:00
Cargo.toml
Add a custom softmax implementation. (#744)
2023-09-05 14:20:23 +01:00
README.md
Add some missing readme files. (#304)
2023-08-02 10:57:12 +01:00

README.md

candle-nn