candle/candle-nn
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 Add some missing readme files. (#304) 2023-08-02 10:57:12 +01:00

README.md

candle-nn