Laurent Mazare ffeafbfc43 Make the nll op closer to the pytorch version + add a test. (#286)

2023-07-31 14:14: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

Improve the mnist training example. (#276)

2023-07-29 16:28:22 +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

Improve the mnist training example. (#276)

2023-07-29 16:28:22 +01:00

linear.rs

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

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

loss.rs

Make the nll op closer to the pytorch version + add a test. (#286)

2023-07-31 14:14:01 +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

More mnist training. (#275)

2023-07-29 13:29:31 +01:00