Laurent Mazare 4b3bd79fbd Remove the embedding ops in favor of index-select. (#299) ...

* Remove the embedding ops in favor of index-select.

* Also remove the cuda kernels.

2023-08-02 05:42:11 +01:00

..

encodec_model.rs

Remove the embedding ops in favor of index-select. (#299)

2023-08-02 05:42:11 +01:00

main.rs

Add some 'cuda-if-available' helper function. (#172)

2023-07-15 08:25:15 +01:00

musicgen_model.rs

Softmax numerical stability. (#267)

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

nn.rs

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

2023-07-14 08:22:08 +01:00

t5_model.rs

Softmax numerical stability. (#267)

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