https://github.com/spacetelescope
The Physical Optics Propagation in PYthon (Poppy) has some neat pictures: https://poppy-optics.readthedocs.io/en/latest/