I'm a PhD candidate in computational cosmology at the AstroParticule & Cosmology Laboratory (APC, CNRS/IN2P3) in Paris, building differentiable and distributed cosmological simulations for the next generation of cosmological surveys.
- Full-field weak lensing inference - Developing scalable simulation pipelines for LSST
- CMB component separation - Part of the Simons Observatory collaboration
- Cross-survey cosmology - Combining CMB and large-scale structure datasets for enhanced constraints
- High-performance computing - Leveraging JAX, CUDA, and distributed computing for cosmological modeling
| Project | Description |
|---|---|
| jaxDecomp | JAX bindings to NVIDIA cuDecomp for distributed 3D FFTs and multi-GPU domain decomposition |
| JaxPM | Particle-mesh cosmological simulation toolkit in JAX with multi-accelerator support |
| FURAX | Framework for Unified and Robust data Analysis with JAX for inverse problems in cosmology |
| jax-healpy | JAX-native HEALPix utilities - GPU-ready and differentiable |
Specialties: Distributed Computing | Automatic Differentiation | Bayesian Inference | HPC | CMB Analysis | Weak Lensing
Website: askabalan.github.io