google-deepmind/dm_pix
PIX is an image processing library built on JAX with GPU/TPU optimization via JAX's jit, vmap, and pmap.
Not currently ranked — collecting fresh signals.
star history
PIX provides image processing functions optimized for JAX, enabling parallelized and JIT-compiled operations for ML research workflows. It wraps common image transformations (flip, resize, crop, color adjustments) in JAX-compatible functions that integrate with JAX’s automatic differentiation and accelerator support.
Frequently asked
- What is google-deepmind/dm_pix?
- PIX is an image processing library built on JAX with GPU/TPU optimization via JAX's jit, vmap, and pmap.
- Is dm_pix open source?
- Yes — google-deepmind/dm_pix is open source, released under the Apache-2.0 license.
- What language is dm_pix written in?
- google-deepmind/dm_pix is primarily written in Python.
- How popular is dm_pix?
- google-deepmind/dm_pix has 441 stars on GitHub.
- Where can I find dm_pix?
- google-deepmind/dm_pix is on GitHub at https://github.com/google-deepmind/dm_pix.