huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
96a48e5cc42b3c94d9d9687bb29987953df36db8
candle/candle-core
History
Laurent Mazare 96a48e5cc4 Add argsort. (#2132) 
* Add the argsort cuda kernels.

* CPU version of arg-sort.

* Hook the cuda kernel + rework the cpu bits.

* Add some dedicated test.

* Working cuda kernel.

* Metal kernel.

* Metal adjustments.

* Bugfix.

* Use the fast rope in qwen.

* Rework the expert selection in qwen.
2024-04-27 20:17:35 +02:00
..
benches
Metal Unary: Add benchmarks and process kernels in a tile based fashion (#2056)
2024-04-21 00:10:33 +02:00
examples
Move the tensor-tools binary in a separate crate. (#1969)
2024-03-30 15:49:37 +01:00
src
Add argsort. (#2132)
2024-04-27 20:17:35 +02:00
tests
Add argsort. (#2132)
2024-04-27 20:17:35 +02: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