Laurent Mazare 620f83cf66 Add the candle-datasets crate (#322) ...

* Move the vision datasets to a separate crate.

* Move the batcher bits.

* Update the readme.

* Move the tiny-stories bits.

---------

Co-authored-by: Jane Doe <jane.doe@example.org>

2023-08-05 08:56:50 +01:00

..

activation.rs

[nn] Move the Embedding and Activation parts. (#116)

2023-07-10 10:24:52 +01:00

conv.rs

Rename the .r functions to .dims so as to be a bit more explicit. (#220)

2023-07-22 10:39:27 +01:00

embedding.rs

Use index-select for the embeddings as it supports backprop. (#298)

2023-08-01 20:44:43 +01:00

init.rs

Improve the mnist training example. (#276)

2023-07-29 16:28:22 +01:00

layer_norm.rs

Llama more training (#297)

2023-08-01 19:53:41 +01:00

lib.rs

Add the candle-datasets crate (#322)

2023-08-05 08:56:50 +01:00

linear.rs

Llama more training (#297)

2023-08-01 19:53:41 +01:00

loss.rs

Add the cross-entropy loss. (#287)

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

ops.rs

Softmax numerical stability. (#267)

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

optim.rs

Add the AdamW optimizer. (#307)

2023-08-02 14:03:49 +01:00

var_builder.rs

Llama more training (#297)

2023-08-01 19:53:41 +01:00