feat: unify edge_degree + layer radial MLPs into single batched GEMM#1849

misko wants to merge 3 commits intomainfacebookresearch/fairchem:mainfrom

unified_radial_edge_degreefacebookresearch/fairchem:unified_radial_edge_degreeCopy head branch name to clipboard