Summary
The proposed thesis aims to develop and apply a probabilistic programming language written in JAX and using SPMD parallelism on GPUs to scale Bayesian inference tasks on large cosmological datasets. This language has no equivalent on the market and will be a major asset for the scientific community. The thesis will require the development of low-level code and will be led collaboratively following the open-source philosophy.