QuantifAI
is a PyTorch-based open-source radio interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driven (learned) priors.
The methods developed and implemented in QuantifAI
are also being ported to the C++ PURIFY and SOPT codes for exascale computing.