diff --git a/.github/workflows/validate.yml b/.github/workflows/validate.yml
index d177d18..55f43ad 100644
--- a/.github/workflows/validate.yml
+++ b/.github/workflows/validate.yml
@@ -25,7 +25,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        os: [blaze/macos-14, ubuntu-latest, windows-latest]
+        os: [blaze/macos-14, blaze/ubuntu-latest, windows-latest]
         rust: [stable]
         backend: ["ndarray", "wgpu", "torch"]
         include:
@@ -35,9 +35,6 @@ jobs:
           # windows can have CPU Vulkan but Burn doesn't select CPU well yet
           - os: windows-latest
             backend: "wgpu"
-          # ubuntu is throwing SIGSEGV
-          - os: ubuntu-latest
-            backend: "wgpu"
     steps:
       - name: Checkout
         uses: actions/checkout@v2