Python EquivariantSelfAttention Resources