96 citations · 109 across the 12 of their papers we have counts for
5 papers · 1 filter
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…
Equivariant Message Passing Neural Network for Crystal Material Discovery
Astrid Klipfel, Olivier Peltre, Najwa Harrati +3
Automatic material discovery with desired properties is a fundamental challenge for material sciences. Considerable attention has recently been devoted to generating stable crystal…
Geomstats: A Python Package for Riemannian Geometry in Machine Learning
Nina Miolane, Alice Le Brigant, Johan Mathe +16
We introduce Geomstats, an open-source Python toolbox for computations and statistics on nonlinear manifolds, such as hyperbolic spaces, spaces of symmetric positive definite matri…