I'm a physicist at UNAM working at the intersection of machine learning and scientific computing. My research spans cosmological simulation acceleration, interpretable medical imaging, and differentiable modeling of physical systems — with emphasis on methods that are physically grounded, scalable, and generalizable.
-
🔭🪐 I’m currently working on Machine learning applied to Cosmology
-
🌱 I’m currently learning JAX
-
🌎 I’m looking to collaborate on Machine Learning astrophysics projects
-
⚡ Fun fact I love to do astrophotography and volunteering
- Microsoft Learn Student Ambassaor
"The most incomprehensible thing about the universe is that it is comprehensible."
Open to collaborations in ML for science, cosmology, and medical imaging.
diegoviillalba.github.io · diego.villalba@ciencias.unam.mx