huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
58605252e8c9355d6f2452f54918e9eb4b938b1f
candle/candle-core
History
Thomas Santerre 754fa1e813 Add support for max_pool2d for Metal backend (#1863) 
* first pass at implementation of maxpool2d

* Add definitions for other dtypes

* add tests for other dtypes

* Cosmetic tweaks + re-enable maxpool2d tests for metal.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>
2024-03-18 08:33:30 +01:00
..
benches
Merge branch 'main' into ivarflakstad/metal-prng
2024-01-14 17:10:54 +01:00
examples
Cuda kernel for dequantizing q8k. (#1760)
2024-02-26 08:42:44 +01:00
src
Add support for max_pool2d for Metal backend (#1863)
2024-03-18 08:33:30 +01:00
tests
Add support for max_pool2d for Metal backend (#1863)
2024-03-18 08:33:30 +01:00
Cargo.toml
feat(bf16): add cast support + tests for cast + bin ops (#1524)
2024-01-11 15:49:13 +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