I was just working on a small protein diffusion model and felt bad when I started copying and pasting quaternion functions from pytorch3d to avoid dependency he

I was just working on a small protein diffusion model and felt bad when I started copying and pasting quaternion functions from pytorch3d to avoid dependency hell. Lo and behold I see Chai did the same shit in their repo. Lol