Inspiration: https://github.com/lucidrains/denoising-diffusion-pytorch https://github.com/lucidrains/bit-diffusion https://github.com/lucidrains/protein-bert-pytorch