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

..

examples

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

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

src

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

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

build.rs

Add flash attention (#241)

2023-07-26 07:48:10 +01:00

Cargo.toml

Add training for the llama2.c example (#296)

2023-08-01 17:23:07 +01:00