Super-resolution microscopy - JAX