Taking Equivariance in deep learning for a spin?

I recently went to Sheh Zaidi‘s brilliant introduction to Equivariance and Spherical Harmonics and I thought it would be useful to cement my understanding of it with a practical example. In this blog post I’m going to start with serotonin in two coordinate frames, and build a small equivariant neural network that featurises it.

I then realised that although my little python notebook rendered fine in my editor view on wordpress, it doesn’t render at all when published. I would therefore like to redirect you directly to the notebook’s gist. Perhaps in future I will figure this out and write a blog post about embedding things in blog posts. How meta. Enjoy!

Author