Pretrain code

Thanks so much @loganesian411 and @Savaw for making your code open source, and congrats on the great paper!
I was wondering if it would be possible to make the pretraining code available? I was searching in the repo and unfortunately didn't find the pretrain functions.