Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
FlashCart accelerates equivariant interatomic potentials by efficiently computing higher-order Cartesian tensor products via generated GPU kernels and recursive feature compression. This approach enables the representation of more complex local correlations, leading to significant accuracy improvements. A FlashCart model with 5.6M parameters achieves superior accuracy and 10x faster inference than a 189M parameter transformer on SPICE-MACE-OFF.
Equivariant potentials can achieve transformer-level accuracy with 35x fewer parameters and 10x faster inference by leveraging Cartesian tensor products.
Machine-learned interatomic potentials extend atomistic simulations beyond the length- and timescales accessible to electronic-structure methods. However, the computational cost of equivariant architectures limits the local correlations they can represent in practice and therefore their achievable accuracy. Here we introduce FlashCart, which makes higher-order correlations affordable by combining generated GPU kernels with an architecture that recursively builds equivariant features and compresses them to a fixed width at each step. We express tensor products in independent Cartesian components and symbolically simplify them and their derivatives, producing fused kernels that often outperform optimized spherical counterparts. We then show that increasing correlation order improves accuracy more efficiently than increasing width, depth, or tensor rank. On SPICE-MACE-OFF, FlashCart models advance the measured accuracy-efficiency frontier: a model with $5.6$ million parameters achieves lower energy and force errors and $10\times$ faster inference than a transformer with $189$ million parameters.