Browsing: JAX3D

@jax.jit def render_chunk(params, o, d, rng): _, out_f, aux = render_rays(params, o, d, rng, deterministic=True) return out_f.ray_values[“rgb”], out_f.ray_depth, out_f.ray_alpha, aux…