Add experiments notebooks

.github/workflows/checks.yml

@@ -36,15 +36,31 @@ jobs:
     env:
       WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        versions:
+          - python: "3.10"
+            torch: "1.13.1"
+          - python: "3.10"
+            torch: "2.0.1"
+          - python: "3.11"
+            torch: "2.0.1"
     steps:
       - name: Checkout code
         uses: actions/checkout@v3
 
+      - name: Set up python
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.versions.python }}
+
       - name: Install dependencies
         run: |
           curl -sSL https://install.python-poetry.org | python3 -
           poetry lock --check
-          poetry install
+          export CUDA_VISIBLE_DEVICES=0
+          poetry add torch@${{ matrix.versions.torch }}+cpu --source torch_cpu
+          poetry install --all-extras
 
       - name: Unit tests
         run: make unit

.gitignore

@@ -5,6 +5,7 @@ models/
 wandb
 tests/_temp/**
 tests/**/_temp/**
+notebooks/data/**
 
 .coverage
 htmlcov/