dm-pix (pypi): PIX is an image processing library in JAX, for JAX.
Keywords: computer-vision|image|image-processing|jax|machine-learning|python