4 Commits

Author	SHA1	Message	Date
Laurent Mazare	c950a5c6b1	Cuda support for the mnist training. (#277) ... * Cuda support for the mnist training. * min/max fix + testing. * Add the argmin/argmax tests. * More cuda support for argmin/argmax. * Cuda kernels for argmin and argmax.	2023-07-29 19:48:04 +01:00
Laurent Mazare	536c5e702e	Cuda kernels for fast min/max reductions (#203) ... * Add the min/max cuda kernels. * Better integration of the cuda kernels.	2023-07-19 18:12:27 +01:00
laurent	ec79fc43f2	Add the bf16 cuda kernels.	2023-06-29 23:12:02 +01:00
Nicolas Patry	d7f729fb8f	Refactor the hierarchy.	2023-06-27 11:57:27 +02:00