Laurent Mazare 1064b9b031 Add the cross-entropy loss. (#287)

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

Add the cross-entropy loss. (#287)

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