Implementing a geometric deep learning module from scratch