Initial release with updated docs

Initial release to go with the arxiv preprint: https://arxiv.org/abs/2110.07470