Laurent Mazare 0902846f25 Add the AdamW optimizer. (#307) ...

* Add the AdamW optimizer.

* Add some AdamW test validated against PyTorch.

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

..

backend.rs

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

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

backprop.rs

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

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

conv.rs

Modular backends (#138)

2023-07-11 11:17:02 +01:00

convert.rs

Add the SGD optimizer (#160)

2023-07-13 19:05:44 +01:00

cpu_backend.rs

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

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

cuda_backend.rs

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

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

device.rs

Improve the mnist training example. (#276)

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

display.rs

Remove some dead-code pragmas. (#137)

2023-07-11 09:33:59 +01:00

dtype.rs

Simplify Tensor::randn. (#255)

2023-07-27 07:40:36 +01:00

dummy_cuda_backend.rs

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

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

error.rs

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

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

indexer.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

layout.rs

Broadcasting performance optimization (cpu) (#182)

2023-07-17 13:41:09 +01:00

lib.rs

Add the AdamW optimizer. (#307)

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

mkl.rs

Mklize more unary ops. (#191)

2023-07-18 13:32:49 +01:00

npy.rs

Allow for lazy loading of npz files, use it in llama to reduce memory usage in the cpu version. (#141)

2023-07-11 20:22:34 +01:00

op.rs

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

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

safetensors.rs

Remove map ownership from save.

2023-08-01 17:19:22 +02:00

shape.rs

Complexifying our hello world

2023-08-01 14:26:02 +02:00

storage.rs

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

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

strided_index.rs

Broadcasting performance optimization (cpu) (#182)

2023-07-17 13:41:09 +01:00

tensor.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

utils.rs

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

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

variable.rs

Improve the mnist training example. (#276)

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