Isometric Tensor Network
This project is used for the paper "Differentiable Programming of Isometric Tensor Network". (arXiv:2110.03898)
Main features:
- Auto-gradient based isometric tensor network construction (MERA and TNR).
- Several built-in spin models
- Multiple optimizers and retraction methods
- Compatible with the graduate lifting bond dimension trick
- Dynamical switching optimizers
- Computing scaling dimensions
In future:
- More models, optimizers and retraction methods
- Layers for variation structures of MERA and TNR
- Quantum machine learning
- ...
Dependencies
The code requires Python
and PyTorch
, with optional CUDA support.