Laurent Mazare ded93a1169 Add the SGD optimizer (#160) ...

* Add the nn::optim and some conversion traits.

* Add the backward_step function for SGD.

* Get the SGD optimizer to work and add a test.

* Make the test slighly simpler.

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

..

backend.rs

Implement the backend trait for the cpu backend. (#143)

2023-07-12 09:54:33 +01:00

backprop.rs

Tensor mutability (#154)

2023-07-13 11:04:40 +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

Implement the backend trait for the cpu backend. (#143)

2023-07-12 09:54:33 +01:00

cuda_backend.rs

Modular backends (#138)

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

device.rs

Implement the backend trait for the cpu backend. (#143)

2023-07-12 09:54:33 +01:00

display.rs

Remove some dead-code pragmas. (#137)

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

dtype.rs

Add from_iter and arange, use it in the doctests. (#145)

2023-07-12 12:03:01 +01:00

dummy_cuda_backend.rs

Modular backends (#138)

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

error.rs

Move the variable creation to the variable module. (#159)

2023-07-13 16:55:40 +01:00

indexer.rs

Test the index op + bugfix. (#148)

2023-07-12 15:42:36 +01:00

layout.rs

Cleanup the main crate error and add a couple dedicated ones (#142)

2023-07-12 09:17:08 +01:00

lib.rs

Add the SGD optimizer (#160)

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

mkl.rs

Remove the dependency to blas and use mkl directly. (#125)

2023-07-10 15:52:03 +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

Add the elu op. (#113)

2023-07-09 21:56:31 +01:00

safetensors.rs

Tensor mutability (#154)

2023-07-13 11:04:40 +01:00

shape.rs

Tensor mutability (#154)

2023-07-13 11:04:40 +01:00

storage.rs

Modular backends (#138)

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

strided_index.rs

Get the cpu tests to run.

2023-06-28 14:32:02 +01:00

tensor.rs

Move the variable creation to the variable module. (#159)

2023-07-13 16:55:40 +01:00

utils.rs

Resurrect the llama npy support. (#140)

2023-07-11 19:32:10 +01:00

variable.rs

Add the SGD optimizer (#160)

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