chromatix (pypi): Differentiable computational optics library using JAX!
Keywords: differentiable|simulation|computational optics|machine learning