Make sure you have python 3.11 installed.
First, ensure you have uv installed. You can install it via pip:
pip install uvThen, install all dependencies specified in pyproject.toml and uv.lock:
uv pip install -r pyproject.tomlYou might need to check the PyTorch compatibility with your CUDA version. Information available here: https://pytorch.org/get-started/
Start Jupyter Lab or Notebook:
uv run -- jupyter labor
uv run -- jupyter notebookor You can also activate the virtualenv manually:
source .venv/bin/activateand then run jupyter
Open experiments.ipynb and run all cells to reproduce the results.
Note: All dependencies are pinned in uv.lock, versions can be found in pyproject.toml.