Module gpu_kernels

Source
Expand description

A collection of GPU kernels for common operations

Functionsยง

add
Add two arrays element-wise.
matmul
Matrix multiplication.
multiply
Multiply two arrays element-wise.