Module ops 

Implementation of common array operations with automatic device selection.

Functions

add

Element-wise addition with automatic device selection.

matmul

Matrix multiplication with automatic device selection.

multiply

Element-wise multiplication with automatic device selection.