Unrelated to the technical discussion but I was wondering what you made that architecture gif with? Looks neat!

I think that image is from the paper and was not created by me. Looks cool indeed!