AFAIK, please bear with my little knowledge. It should probably works. In the background library like Tensorflow can handle any matrix and tensor operations easily by using GPU or TPU. Google support both of them.

This is basic example for any linear algebra operations using tensorflow library. Hope this will help you.