60 citations · 295 across the 19 of their papers we have counts for
4 papers · 2 filters
E3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPU
Olivier Peltre, Armand Picard, Adrien Pichard +6
We present e3j, a fast Euclid-equivariance backend for geometric deep learning applications with JAX bindings for GPU and TPU. Leveraging both optimized CUDA and Pallas kernels and…
Force-Aware Neural Tangent Kernels for Scalable and Robust Active Learning of MLIPs
Eszter Varga-Umbrich, Zachary Weller-Davies, Paul Duckworth +3
Active learning for machine-learning interatomic potentials (MLIPs) must address several challenges to be practical: scaling to large candidate pools, leveraging energy-force super…
Pretrained Model Representations as Acquisition Signals for Active Learning of MLIPs
Eszter Varga-Umbrich, Shikha Surana, Paul Duckworth +3
Training machine learning interatomic potentials (MLIPs) for reactive chemistry is often bottlenecked by the high cost of quantum chemical labels and the scarcity of transition sta…
Integral Formulas for Vector Signal Tensor Products
Valentin Heyraud, Zachary Weller-Davies, Jules Tilly
We derive integral formulas that simplify the Vector Signal Tensor Product recently introduced by Xie et al., which generalizes the Gaunt tensor product to anti-symmetric couplings…