Diffusion Models for Gravitational Lensing

Diffusion Models for Gravitational Lensing Simulation

Developed a diffusion model in PyTorch for gravitational lensing simulation, optimized with UNet architectures and GPU memory improvements to handle large-scale simulations.

Skills: PyTorch, Diffusion Models, UNet