OTF Gym: A Set of Reinforcement Learning Environment of Layered Optical Thin Film Inverse Design
An-Qing Jiang, Liang‐Yao Chen, Osamu Yoshie · Conference on Lasers and Electro-Optics · 2021
OTF gym is a python library for optical thin film inverse design. This paper focuses on creating a collection of the reinforcement learning environment to achieve a multi-layer optical thin film inverse design. We develop environments based on TMM simulation to solve various op- timization targets(material, thickness, structural) with a single-agent and multi-agent reinforcement learning algorithm. This code is available in Github.