Laurent Mazare e3db30021f Support for "unbatched" rope. (#2926) ...

* Support for (un)-batched rope.

* Use 3d rope in the rope/ropei/rope_thd functions.

* Get the CPU versions to work.

* Fix the cuda version.

* Adapt the metal side.

* Fix the metal tests.

2025-04-27 15:12:02 +02:00

..

batch_norm.rs

Bug Fix: When converting a tensor to a variable, clone if the tensor is already a variable. (#2124)

2024-04-29 11:21:53 +02:00

group_norm.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

kv_cache.rs

Rotating kv cache positions (#2901)

2025-04-15 23:09:26 +02:00

layer_norm.rs

Enable the new layer-norm. (#2213)

2024-05-24 16:48:21 +02:00

loss.rs

Add Binary Cross Entropy With Logit Loss to nn crate (#1157)

2023-10-23 17:12:44 +01:00

one_hot.rs

Add one-hot/cold encoding (#1489)

2024-01-01 11:18:40 +01:00

ops.rs

Support for "unbatched" rope. (#2926)

2025-04-27 15:12:02 +02:00

optim.rs

Tweak the VarMap set type. (#1758)

2024-02-25 20:50:08 +01:00

rnn.rs

Add a GRU layer. (#688)

2023-08-31 08:43:10 +01:00

sdpa.rs

Make the metal sdpa tests deterministic. (#2750)

2025-01-28 09:05:24 +01:00