Software

CosmoFuse

A highly parallelized, GPU-accelerated Python framework for measuring the integrated 3-point correlation function directly from simulated weak lensing maps on HEALPix grids.

• JAX-based backend for efficient GPU computation
• Operates directly on HEALPix convergence and shear maps
• Used in the SBi3PCF pipeline to process ~10⁵ mock observations

i3PCF

Analytical predictions for the integrated 3-point correlation function of projected cosmic density fields. I contributed implementations of observational effects (reduced shear, source clustering) into the theory code.

affinax

An affine-invariant ensemble MCMC sampler implemented in JAX.

emax

A JAX-based implementation of a simple emulator architecture.

cosm_parameter_mcmc

MCMC reconstruction of the posterior for the alpha-DGP cosmological model fitted to supernovae observations. Written in C using GSL.

SBI Tutorial

A tutorial for machine learning methods and simulation-based inference.