huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 04:00:28 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,367 Commits 74 Branches 16 Tags
cd96fa80da255e34f7b16b4ff98b6a31d557201b
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
9a5c7db91a Add support for i64 (#563) 
* Add the i64 dtype.

* Adapt the cuda kernels.
 2023-08-23 10:42:19 +01:00
Laurent Mazare
e12372021b Expose the tensor write-bytes function. (#412) 2023-08-11 17:13:42 +01:00
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