huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
a0460cd2b13a396ff8545dc1bbffa741f2ec3d79
candle/candle-nn
History
Laurent Mazare 2ac302a5d1 Add the rope THD kernel. (#2014) 
* Add the rope THD kernel.

* Cuda kernel for rope-thd.

* Add the metal kernels.

* Add a dedicated test.
2024-04-05 08:32:58 +02:00
..
benches
Add benchmarks for the candle-nn package (#1995)
2024-04-03 07:03:54 +02:00
examples
Optimize the cat operation on contiguous tensors (#1855)
2024-03-17 10:49:13 +01:00
src
Add the rope THD kernel. (#2014)
2024-04-05 08:32:58 +02:00
tests
Add the rope THD kernel. (#2014)
2024-04-05 08:32:58 +02:00
Cargo.toml
Add benchmarks for the candle-nn package (#1995)
2024-04-03 07:03:54 +02:00
README.md
…

README.md

candle-nn