jax-based-optical-tools
TMMax: High-performance modeling of multilayer thin-film structures using transfer matrix method with JAX