JAX is a library resulting from the union of Autograd and XLA for high-performance machine learning research. It provides NumPy, SciPy, automatic differentiation and first-class GPU/TPU support.
PIX is an image processing library built don top of JAX, for JAX. Its goal is providing image processing functions and tools to JAX in a way that they can be optimised and parallelised through `jax.jit`, `jax.vmap` and `jax.pmap`.
