Laurent Mazare 1c9e5394a5 Add a custom softmax implementation. (#744) ...

* Add a custom softmax implementation.

* Add softmaxlastdim to the benchmarks.

* And add a test.

* Support more dtypes.

* Polish the code.

* Use the slow implementation on cuda.

* Add a todo for the cuda kernel.

2023-09-05 14:20:23 +01:00

..

batch_norm.rs

Avoid some redundant clone. (#731)

2023-09-04 09:18:32 +02:00

group_norm.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

layer_norm.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

loss.rs

Move the test-utils bits to a shared place. (#619)

2023-08-27 09:42:22 +01:00

ops.rs

Add a custom softmax implementation. (#744)

2023-09-05 14:20:23 +01:00

optim.rs

Add the optimizer trait. (#702)

2023-09-01 12:55:39 +01:00

rnn.rs

Add a GRU layer. (#688)

2023-08-31 08:43:10 +01:00