Laurent Mazare 2345b8ce3f Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337) ...

* Skeleton for the avg-pool2d and upsample-nearest2d ops.

* Preliminary conv2d support.

2023-08-07 16:15:38 +01:00

..

activation.rs

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

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

conv.rs

Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337)

2023-08-07 16:15:38 +01:00

embedding.rs

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

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

group_norm.rs

Implement group-norm. (#334)

2023-08-07 06:53:05 +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 a stable diffusion example (#328)

2023-08-06 17:49:43 +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

Implement group-norm. (#334)

2023-08-07 06:53:05 +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