This work introduces ElemeNet, a unified, general-purpose software package for molecular machine learning. The ElemeNet software package enables the training of advanced ML models for diverse properties and data sets with an enlarged range of elemental compositions.

We define molecular representations compatible with elements 1–100, supporting diverse organometallic and biological systems in addition to organic chemistry already well served by the Chemprop ML toolkit. As well as more common atom-, bond-, and molecule-level predictions, we introduce and allow for moiety-level predictions.

We also natively define optional conditioning on charge and spin states. Advanced E(3)-equivariant and transformer architectures are supported in addition to 2D models, with all classes including built-in uncertainty quantification through deterministic and statistical measures. We benchmark our protocols for ML model training against representative data sets from organic, inorganic, coordination, and biological chemistry, achieving competitive and SOTA performance relative to literature baselines and favorable scaling to millions of molecules.