Bioprocess Modeling Made Easy With Hybrax
Hybrid models that embed neural networks in ordinary differential equations are increasingly popular in bioprocess modeling, but they are challenging to set up. Much of the effort goes into the code around the model: wet-lab data must be imported and pre-processed, feeds and sampling must be accounted for in the balance equations, and a training loop with cross-validation must be set up. Researchers usually write this code from scratch for each project. In addition, no widely adopted format for bioprocess data exists, and datasets are mostly published in bespoke spreadsheets or not at all. As a result, models and datasets from the literature are difficult to reproduce, reuse, or compare.We present Hybrax, an open-source Python framework that addresses these problems. It combines a flexible format for bioprocess data with a customizable end-to-end training pipeline for mechanistic and hybrid models. Hybrax derives the transport terms in the process dynamics automatically and provides the utilities needed for training, so that researchers can focus on developing the model itself. It also facilitates model sharing and reproducibility: each training checkpoint contains the model parameters, code, configuration, and training data required to run the model or retrain it from scratch. We demonstrate Hybrax by re-implementing two models from the literature, one trained on real-world E. coli data and one on synthetic HEK293 data, with performance comparable to the original models. Hybrax is available at https://github.com/hybrax/hybrax.