Geometric deep learning for functional protein design