This is placeholder. The repository will be populated once the article will be submitted for archival publication.
In the meanwhile you can take a look at the poster presented at Machine Learning and the Physical Sciences workshop at NeurIPS 2022.
The data used for creating the plots can be found here: https://osf.io/htcvs/ (slowly updated)