huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 19:58:35 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
a094dc503d69a6ca3db71098ebc26d0d2f2a33a6
candle/candle-core
History
Laurent Mazare a094dc503d Add a cuda kernel for avg-pool2d. (#440) 
* Add a cuda kernel for avg-pool2d.

* Avoid running out of bounds.

* Finish wiring the avg pool kernel + add some testing.

* Support for max-pool + testing.
2023-08-14 12:32:05 +01:00
..
examples
Add a softmax bench. (#433)
2023-08-13 20:09:18 +01:00
src
Add a cuda kernel for avg-pool2d. (#440)
2023-08-14 12:32:05 +01:00
tests
Add a cuda kernel for avg-pool2d. (#440)
2023-08-14 12:32:05 +01:00
Cargo.toml
Parallelise the CPU kernels for the conv ops. (#401)
2023-08-11 05:51:58 +01:00
LICENSE
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00
README.md
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00

README.md

candle

Minimalist ML framework for Rust