Include an example of getting image back as JAX array

Author: horizon-blue

tests/python_tests/test_forward.py

@@ -40,3 +40,5 @@ def test_render_fmbs_smoke(blender, confidence) -> None:
 
     assert img_view.num_rows == camera.height
     assert img_view.num_cols == camera.width
+    depth_image = img_view.depth.as_jax()
+    assert isinstance(depth_image, jnp.ndarray)