Hi @Geegee, welcome to the forum 
In principle, yes. Nx is designed to work on the GPU and in fact has a least squares function built in:
The trouble will be getting the data out of the .npy format. A quick search turned up this library:
That appears to be exactly what you want. I have not tried that library though so I can’t vouch for it.
