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

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

..

examples

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

2023-07-31 14:14:01 +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 a flag to save the trained weights. (#279)

2023-07-30 15:41:42 +01:00