PIX is an image processing library in JAX, for JAX. - View it on GitHub