use uv in gh actions (#192)

* use uv for speeding up pip install during test

Author: lucidrains

.github/workflows/test.yml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        group: [0, 1, 2, 3, 4, 5, 6, 7]
+        group: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
 
     steps:
     - uses: actions/checkout@v4
@@ -23,9 +23,10 @@ jobs:
         python-version: "3.10"
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
-        python -m pip install torch==2.3.1
-        python -m pip install --default-timeout=100 -e .[test]
+        python -m pip install uv
+        python -m uv pip install --upgrade pip
+        python -m uv pip install torch==2.3.1
+        python -m uv pip install -e .[test]
     - name: Test with pytest
       run: |
-        python -m pytest --num-shards 8 --shard-id ${{ matrix.group }} tests/
+        python -m pytest --num-shards 10 --shard-id ${{ matrix.group }} tests/