Related paper: https://arxiv.org/abs/2306.11054
Content of this repository:
MLRG.py
contains the PyTorch implementation of the MLRG algorithm.MLRG.ipynb
contains code to train the models.MLRG.nb
contains code to analyze the trained models and extract physical properties.- Models are saved in the folder
models
.