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