Introduction

Equilibrium model

summary of Z-T paper up to the point of the equilibrium model, with some details on the model itself

Dataset generation

explaination and script documentation for the dataset generation

Training

training sripts available, with some details on the training procedure and hyperparameters