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

2023-07-31 14:14:01 +01:00

..

layer_norm.rs

Simplify the parameters used by sum and sum_keepdim. (#165)

2023-07-14 08:22:08 +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 pytorch version of the linear regression as a comment. (#163)

2023-07-13 21:05:57 +01:00