{"repo":"google-deepmind/dm_pix","free":true,"listed":false,"github":"https://github.com/google-deepmind/dm_pix","clone":"git clone https://github.com/google-deepmind/dm_pix.git","description":"PIX is an image processing library in JAX, for JAX.","language":"Python","stars":444,"topics":["jax","image","image-processing","machine-learning","computer-vision","python"],"license":"Apache-2.0","category":"machine-learning","readme_excerpt":"PIX PIX is an image processing library in [JAX], for [JAX]. Overview [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 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 ][jit], [ jax.vmap ][vmap] and [ jax.pmap ][pmap]. Installation PIX is written in pure Python, but depends on C++ code via JAX. Because JAX installation is different depending on your CUDA version, PIX does not list JAX as a dependency in [ pyproject.toml ], although it is technically listed for reference, but commented. First, follow [JAX installation instructions] to install JAX with the relevant accelerator support. Then, install PIX using pip : Quickstart To use PIX , you just need to import dm pix as pix and use it right away! For example, let's assume to have loaded the JAX logo (available in examples/assets/jax logo.jpg ) in a variable called image and we want to flip it left to right. ![JAX logo] All it's needed is the following code! And here is the result! ![JAX logo left-right] All the functions in PIX can be [ jax.jit ][jit]ed, [ jax.vmap ][vmap]ed and [ jax.pmap ][pmap]ed, so all the following functions can take advantage of optimization and parallelization. You can check it yourself that the result from the four versi","default_branch":null,"files":null,"tree":[],"storefront":"/r/google-deepmind","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/google-deepmind/dm_pix/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}