Laurent Mazare 3eb2bc6d07 Softmax numerical stability. (#267) ...

* Softmax numerical stability.

* Fix the flash-attn test.

2023-07-28 13:13:01 +01:00

..

vision

Vision dataset (#179)

2023-07-16 23:43:55 +01:00

activation.rs

[nn] Move the Embedding and Activation parts. (#116)

2023-07-10 10:24:52 +01:00

conv.rs

Rename the .r functions to .dims so as to be a bit more explicit. (#220)

2023-07-22 10:39:27 +01:00

embedding.rs

Add some documentation and test to the linear layer. (#151)

2023-07-12 20:24:23 +01:00

init.rs

Sketch the tensor initialization module. (#134)

2023-07-11 07:41:46 +01:00

layer_norm.rs

Rename the .r functions to .dims so as to be a bit more explicit. (#220)

2023-07-22 10:39:27 +01:00

lib.rs

Move some shared functions to the nn module. (#221)

2023-07-22 13:25:11 +01:00

linear.rs

Add some documentation and test to the linear layer. (#151)

2023-07-12 20:24:23 +01:00

loss.rs

Move some shared functions to the nn module. (#221)

2023-07-22 13:25:11 +01:00

ops.rs

Softmax numerical stability. (#267)

2023-07-28 13:13:01 +01:00

optim.rs

Add the gradient for reduce-sum. (#162)

2023-07-13 20:14:10 +01:00

var_builder.rs

Added comment about offsets.

2023-07-27 20:11:57 +02:00