D-Flat is a forward and inverse design framework for flat optics. Although specially geared for the design of metasurfaces, it may be used broadly for end-to-end imaging and sensing task.
Open-source Python package for design, analysis, and optimization of programmable electromagnetic metasurfaces (RIS/IRS) for wireless communication and sensing