PyCon UK 2025

Kolen Cheung

Kolen Cheung is a Research Software Engineer with a PhD in Physics (UC Berkeley) specializing in CMB data analysis with a designated emphasis ("graduate minor") in Computational & Data Science & Engineering. Dr. Cheung has contributed to multiple CMB collaborations including POLARBEAR, LiteBIRD, CMB-S4, and served as Software Deployment Lead at Simons Observatory. His current work spans gravitational lensing, quantum biological modeling, and HPC systems, combining theoretical physics with advanced computational methods.


Session

09-21
12:00
25min
JIT compilers for scientific computing in Python: Numba vs. JAX
Kolen Cheung

Accelerating scientific Python with JITs. We share our journey migrating a gravitational lensing likelihood calculation from Numba to JAX. Learn about performance gains, automatic differentiation benefits, and practical lessons for high-performance scientific computing in Python.

Space 2