arXiv cs.AI / cs.LG / cs.CL·3d agoE3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPU#e3j#equivariant-networks#jaxAI tools & infra