preprint release

This is the first code release for the preprint: https://arxiv.org/abs/2208.05424, currently under review