JAX, a library that provides NumPy, SciPy, and automatic differentiation, has become increasingly popular among ML researchers in recent months. Now, Deepmind has released PIX, an image processing library purpose-built for JAX. The library also includes a test suite and a set of examples.