Currently, our CI test suite only tests CPU-only code. Any models that require a GPU/cuda need to be tested manually via Google Colab. A much better way to run these tests would be to figure out a way to configure GitHub Actions to spin up CUDA-enabled machines to run tests too.