jax-based-optical-tools
A fast transfer matrix method written in jax for modelling optical multilayer thin films