Tensor Network Notes

This repo contains my notes and code related to the course Tensor Network.

Environment Setup

This is a uv project. Setting up the environment is easy:

1. If you have not got uv installed, follow the instructions to install it.
2. Run uv sync to create an environment and get dependencies, including development dependencies that you need to run the code in a notebook.

Run Tools

We use poe to run tools. Available commands are:

• lab: run Jupyter Lab
• sync: sync the code in notebooks to the tensor_network package
• format: format the code in notebooks
• Checking: use ruff linter to check code
  • check_tensor_network: check the exported code in tensor_network package
  • check: check code in .
  • check_all: check code in . and tensor_network
• precommit: run pre-commit hooks

Sample usage:

# if your shell detects venvs automatically, you can run poe directly
poe lab
# if your shell does not detect venvs automatically, you can run uv run poe directly
uv run poe lab

Contribution

Contributions are very welcome. Please file an issue or PR if you have any questions or suggestions.

A few points to note:

• The code should primarily live in notebooks, not Python scripts. We use nbdev to export useful code from notebooks to tensor_network package for reusability.

Acknowledgements

• Big thanks to Prof. Ran for the course
• Thanks to Gemini 2.0 for transcribing a lot of equations