Based on JAX, Chromatix provides a standardized framework for developing, sharing and reusing wave-optics simulations. The capabilities are demonstrated in a number of computational microscopy applications.