Add linear algebra tests to our CI workflow
The linear algebra tests check that basic matrix operations work on
AMD GPUs. We've found that these catch a lot of basic issues with the
JAX plugin and the tests don't take long to run, so they'd be useful in
CI.
