diff --git a/.cirun.yml b/.cirun.yml
new file mode 100644
index 0000000..da6059d
--- /dev/null
+++ b/.cirun.yml
@@ -0,0 +1,7 @@
+runners:
+  - name: aws-gpu-runner
+    cloud: aws
+    instance_type: g4dn.xlarge
+    machine_image: ami-00ac0c28c01352e53
+    labels:
+      - cirun-aws-gpu
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 318c2f6..50e089a 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -7,12 +7,14 @@ on:
 
 jobs:
   build:
-    runs-on: ubuntu-latest
+    runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false
       matrix:
         python-version: ["3.7", "3.8", "3.9"]
         pytorch-version: ["1.4.0", "1.5.1", "1.6.0", "1.7.1", "1.8", "1.9", "1.10", "1.11", "1.12"]
+        os:
+          - ubuntu-latest
         exclude:
           - python-version: 3.7
             pytorch-version: 1.11
@@ -24,6 +26,10 @@ jobs:
             pytorch-version: 1.5.1
           - python-version: 3.9
             pytorch-version: 1.4.0
+        include:
+          - python-version: 3.9
+            pytorch-version: 1.12
+            os: cirun-aws-gpu
 
     steps:
     - uses: actions/checkout@v2