e3nn
Atom-JEPA: Joint-Embedding Predictive Architecture for 3D Atomistic Systems
Fully-Differentiable Tensor Spherical Harmonics in JAX