Add GPU compatible version of matrix ops #2
Description
To do, add a version of matrix_ops that works well with GPU - for example a jax or CuPy based version.
{{ message }}
To do, add a version of matrix_ops that works well with GPU - for example a jax or CuPy based version.