Geometric deep learning for protein sequence design