6 Commits

Author	SHA1	Message	Date
laurent	f6104c4b64	Add the reduce-sum kernel.	2023-06-26 12:35:26 +01:00
laurent	16f0f5b9d2	Add a cuda kernel for embeddings.	2023-06-26 11:47:57 +01:00
laurent	5276755fb3	Add cuda support for unary ops.	2023-06-22 15:12:59 +01:00
laurent	b8f514d9c6	Add more binary kernels.	2023-06-22 14:07:02 +01:00
laurent	4b1c3405e9	Add a couple cuda kernels from dfdx.	2023-06-22 12:56:29 +01:00
laurent	083ced4428	Integrate the kernels bits.	2023-06-22 09:59:00 +01:00