Laurent Mazare 7f0f83a7c1 Rotating kv cache positions (#2901) ... * Retrieve the current positions for rotating KV caches. * Add the function to the kv cache too. * More testing.		2025-04-15 23:09:26 +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	upgrade half library to fix rand (#2806)	2025-03-14 09:01:54 +01: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