PIX is a library built on top of JAX with the goal of 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.
β AK (@ak92501) July 15, 2021
github: https://t.co/AXgepSMkBZ pic.twitter.com/F80hdhty7n