diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 1d3c69b..ede52f6 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -19,6 +19,8 @@ jobs:
         run: |
           curl -LsSf https://astral.sh/uv/install.sh | sh
           echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+      - name: Create virtual environment
+        run: uv venv
       - name: Install build dependencies
         run: |
           uv pip install build ".[torch,flax]"
diff --git a/.github/workflows/manual_publish.yml b/.github/workflows/manual_publish.yml
index 99f4bbd..0d17a4b 100644
--- a/.github/workflows/manual_publish.yml
+++ b/.github/workflows/manual_publish.yml
@@ -16,6 +16,8 @@ jobs:
         run: |
           curl -LsSf https://astral.sh/uv/install.sh | sh
           echo "$HOME/.cargo/bin" >> $GITHUB_PATH
+      - name: Create virtual environment
+        run: uv venv
       - name: Install Poetry and dependencies
         run: |
           uv pip install poetry ".[torch,flax]"