io/image: Replace PyTinyrenderer with a Pure JAX renderer#367

Open

JoeyTeng wants to merge 6 commits intogoogle:mainfrom JoeyTeng:main

+201-92

Commits on Jun 2, 2023

feat(io/image): move to JaxRenderer backend
JoeyTeng
committed

Commits on Jun 3, 2023

Commits on Jun 4, 2023

refactor(io/image): refactor to enable jit for rendering
JoeyTeng
committed

Commits on Jun 5, 2023

Refactor(io/image): submit render jobs first then copy back in parallel
JoeyTeng
committed

Commits on Jun 18, 2023

Merge branch 'google:main' into main
JoeyTeng
authored