https://arxiv.org/pdf/2210.02186.pdf looks very interesting

Yeah, I'm currently playing with that entire repository of models. Unfortunately the ones that exist in this package don't seem to fully implement their features as far as I can tell? Pulling the individual models apart with their own data loading has been... a chore.

they do work but you have either convert them to pytorch lightning + make your own data loaders, or try to make your dataset conform to theirs