pythonjaxopenmpigaussian-processesvariational-inferenceinformation-theory
iftpwa
Thomas Jefferson National Accelerator Facility (JLab) · July 2023 — July 2025
Developing a declarative framework for smooth non-parametric statistical models using Python, JAX, and OpenMPI with Gaussian Processes and Variational Inference, enabling Bayesian inference over O(10^6) free parameters.