На основе JAX, Chromatix обеспечивает стандартизированную структуру для разработки, обмена и повторного использования симуляций волновой оптики. Возможности демонстрируются в ряде вычислительных микроскопических приложений.