Laurent Mazare 52c5d8c087 Add the gather op. (#219) ...

* Start adding gather.

* Gather cpu implementation + use in simple training.

* Add scatter_add for the gradient of gather.

* Simple cpu implementation of scatter_add.

* Use gather in the simple-training backprop.

2023-07-22 07:21:28 +01:00

..

examples

Add the gather op. (#219)

2023-07-22 07:21:28 +01:00

src

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

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

Cargo.toml

Removing candle-hub internal to extract into hf-hub standalone.

2023-07-19 15:04:38 +02:00