Theoretical and Computational Chemistry

NEXTorch: A Design and Bayesian Optimization Toolkit for Chemical Sciences and Engineering

Yifan Wang University of Delaware


Automation and optimization of chemical systems require well-inform decisions on what experiments to run to reduce time, materials, and/or computations. Data-driven active learning algorithms have emerged as valuable tools to solve such tasks. Bayesian optimization, a sequential global optimization approach, is a popular active-learning framework. Past studies have demonstrated its efficiency in solving chemistry and engineering problems. We introduce NEXTorch, a library in Python/PyTorch, to facilitate laboratory or computational design using Bayesian optimization. NEXTorch offers fast predictive modeling, flexible optimization loops, visualization capabilities, easy interfacing with legacy software, and multiple types of parameters and data type conversions. It provides GPU acceleration, parallelization, and state-of-the-art Bayesian Optimization algorithms and supports both automated and human-in-the-loop optimization. The comprehensive online documentation introduces Bayesian optimization theory and several examples from catalyst synthesis, reaction condition optimization, parameter estimation, and reactor geometry optimization. NEXTorch is open-source and available on GitHub.


Thumbnail image of Nextorch_manuscript_v3.pdf
download asset Nextorch_manuscript_v3.pdf 0.67 MB [opens in a new tab]

Supplementary material

Thumbnail image of Nextorch_SI_v1.docx
download asset Nextorch_SI_v1.docx 0.37 MB [opens in a new tab]
Nextorch SI v1