probcomp · sritchie · Jul 23, 2024 · Jul 23, 2024 · Jul 23, 2024 · Jul 23, 2024
diff --git a/.gitignore b/.gitignore
@@ -13,5 +13,6 @@ assets/ycbineoat/*
 **/**.json
 **/**.npy
 **/**.txt
+!src/b3d/renderer/**/**.txt
 assets/kitti/*
 **/**.pth
diff --git a/scripts/make_docs.sh b/scripts/make_docs.sh
@@ -1 +1 @@
-pdoc --html b3d --force
+pdoc --html src/b3d --force
diff --git a/setup.py b/setup.py
@@ -12,7 +12,8 @@
 setuptools.setup(
     name="b3d",
     version="0.0.1",
-    packages=setuptools.find_packages(),
+    package_dir={"": "src"},
+    packages=setuptools.find_packages(where="src"),
     package_data={
         "b3d": [
             "nvdiffrast/common/*.h",
@@ -61,7 +62,7 @@
         "fire==0.6.0",
         "torch==2.2.2",
         "torchvision==0.17.2",
-        "jax[cuda12]==0.4.28",
+        "jax[cuda12]~=0.4.28",
         "natsort",
         "distinctipy",
         "pyliblzfse",
@@ -79,5 +80,5 @@
             "b3d_push = b3d.bucket_utils.b3d_push:upload_to_bucket",
         ]
     },
-    python_requires=">=3.6",
+    python_requires=">=3.10",
 )
diff --git a/b3d/__init__.py → src/b3d/__init__.py b/b3d/__init__.py → src/b3d/__init__.py
diff --git a/b3d/bayes3d/__init__.py → src/b3d/bayes3d/__init__.py b/b3d/bayes3d/__init__.py → src/b3d/bayes3d/__init__.py
diff --git a/b3d/bayes3d/enumerative_proposals.py → src/b3d/bayes3d/enumerative_proposals.py b/b3d/bayes3d/enumerative_proposals.py → src/b3d/bayes3d/enumerative_proposals.py
diff --git a/b3d/bayes3d/likelihoods.py → src/b3d/bayes3d/likelihoods.py b/b3d/bayes3d/likelihoods.py → src/b3d/bayes3d/likelihoods.py
diff --git a/b3d/bayes3d/mesh_library.py → src/b3d/bayes3d/mesh_library.py b/b3d/bayes3d/mesh_library.py → src/b3d/bayes3d/mesh_library.py
diff --git a/b3d/bayes3d/model.py → src/b3d/bayes3d/model.py b/b3d/bayes3d/model.py → src/b3d/bayes3d/model.py
diff --git a/b3d/bayes3d/scene_graph.py → src/b3d/bayes3d/scene_graph.py b/b3d/bayes3d/scene_graph.py → src/b3d/bayes3d/scene_graph.py
diff --git a/b3d/bucket_utils/README.md → src/b3d/bucket_utils/README.md b/b3d/bucket_utils/README.md → src/b3d/bucket_utils/README.md
diff --git a/b3d/bucket_utils/b3d_pull.py → src/b3d/bucket_utils/b3d_pull.py b/b3d/bucket_utils/b3d_pull.py → src/b3d/bucket_utils/b3d_pull.py
diff --git a/b3d/bucket_utils/b3d_push.py → src/b3d/bucket_utils/b3d_push.py b/b3d/bucket_utils/b3d_push.py → src/b3d/bucket_utils/b3d_push.py
diff --git a/b3d/bucket_utils/data_util.py → src/b3d/bucket_utils/data_util.py b/b3d/bucket_utils/data_util.py → src/b3d/bucket_utils/data_util.py
diff --git a/b3d/camera.py → src/b3d/camera.py b/b3d/camera.py → src/b3d/camera.py
diff --git a/b3d/chisight/__init__.py → src/b3d/chisight/__init__.py b/b3d/chisight/__init__.py → src/b3d/chisight/__init__.py
diff --git a/b3d/chisight/dense/__init__.py → src/b3d/chisight/dense/__init__.py b/b3d/chisight/dense/__init__.py → src/b3d/chisight/dense/__init__.py
diff --git a/b3d/chisight/dense/dense_model.py → src/b3d/chisight/dense/dense_model.py b/b3d/chisight/dense/dense_model.py → src/b3d/chisight/dense/dense_model.py
diff --git a/.../dense/dense_only_patch_tracking/model.py → .../dense/dense_only_patch_tracking/model.py b/.../dense/dense_only_patch_tracking/model.py → .../dense/dense_only_patch_tracking/model.py
diff --git a/...nse_only_patch_tracking/patch_tracking.py → ...nse_only_patch_tracking/patch_tracking.py b/...nse_only_patch_tracking/patch_tracking.py → ...nse_only_patch_tracking/patch_tracking.py
diff --git a/...chisight/dense/differentiable_renderer.py → ...chisight/dense/differentiable_renderer.py b/...chisight/dense/differentiable_renderer.py → ...chisight/dense/differentiable_renderer.py
diff --git a/b3d/chisight/dense/likelihoods/__init__.py → ...3d/chisight/dense/likelihoods/__init__.py b/b3d/chisight/dense/likelihoods/__init__.py → ...3d/chisight/dense/likelihoods/__init__.py
diff --git a/...ght/dense/likelihoods/image_likelihood.py → ...ght/dense/likelihoods/image_likelihood.py b/...ght/dense/likelihoods/image_likelihood.py → ...ght/dense/likelihoods/image_likelihood.py
diff --git a/.../dense/likelihoods/kray_outlier_volume.py → .../dense/likelihoods/kray_outlier_volume.py b/.../dense/likelihoods/kray_outlier_volume.py → .../dense/likelihoods/kray_outlier_volume.py
diff --git a/b3d/chisight/dense/likelihoods/krays.py → src/b3d/chisight/dense/likelihoods/krays.py b/b3d/chisight/dense/likelihoods/krays.py → src/b3d/chisight/dense/likelihoods/krays.py
diff --git a/...ht/dense/likelihoods/other_likelihoods.py → ...ht/dense/likelihoods/other_likelihoods.py b/...ht/dense/likelihoods/other_likelihoods.py → ...ht/dense/likelihoods/other_likelihoods.py
diff --git a/...ht/dense/likelihoods/simple_likelihood.py → ...ht/dense/likelihoods/simple_likelihood.py b/...ht/dense/likelihoods/simple_likelihood.py → ...ht/dense/likelihoods/simple_likelihood.py
diff --git a/b3d/chisight/particle_system.py → src/b3d/chisight/particle_system.py b/b3d/chisight/particle_system.py → src/b3d/chisight/particle_system.py
diff --git a/b3d/chisight/patch_tracking.py → src/b3d/chisight/patch_tracking.py b/b3d/chisight/patch_tracking.py → src/b3d/chisight/patch_tracking.py
diff --git a/b3d/chisight/sparse/__init__.py → src/b3d/chisight/sparse/__init__.py b/b3d/chisight/sparse/__init__.py → src/b3d/chisight/sparse/__init__.py
diff --git a/b3d/chisight/sparse/dynamic_gps.py → src/b3d/chisight/sparse/dynamic_gps.py b/b3d/chisight/sparse/dynamic_gps.py → src/b3d/chisight/sparse/dynamic_gps.py
diff --git a/b3d/chisight/sparse/dynamic_sfm.py → src/b3d/chisight/sparse/dynamic_sfm.py b/b3d/chisight/sparse/dynamic_sfm.py → src/b3d/chisight/sparse/dynamic_sfm.py
diff --git a/b3d/chisight/sparse/gps_utils.py → src/b3d/chisight/sparse/gps_utils.py b/b3d/chisight/sparse/gps_utils.py → src/b3d/chisight/sparse/gps_utils.py
diff --git a/b3d/chisight/sparse/hgps_utils.py → src/b3d/chisight/sparse/hgps_utils.py b/b3d/chisight/sparse/hgps_utils.py → src/b3d/chisight/sparse/hgps_utils.py
diff --git a/b3d/chisight/sparse/sparse_gps_model.py → src/b3d/chisight/sparse/sparse_gps_model.py b/b3d/chisight/sparse/sparse_gps_model.py → src/b3d/chisight/sparse/sparse_gps_model.py
diff --git a/...ight/sparse/sparse_model_reality_check.py → ...ight/sparse/sparse_model_reality_check.py b/...ight/sparse/sparse_model_reality_check.py → ...ight/sparse/sparse_model_reality_check.py
diff --git a/b3d/colors.py → src/b3d/colors.py b/b3d/colors.py → src/b3d/colors.py
diff --git a/b3d/io/__init__.py → src/b3d/io/__init__.py b/b3d/io/__init__.py → src/b3d/io/__init__.py
diff --git a/b3d/io/data_loader.py → src/b3d/io/data_loader.py b/b3d/io/data_loader.py → src/b3d/io/data_loader.py
diff --git a/b3d/io/feature_track_data.py → src/b3d/io/feature_track_data.py b/b3d/io/feature_track_data.py → src/b3d/io/feature_track_data.py
diff --git a/b3d/io/mesh_data.py → src/b3d/io/mesh_data.py b/b3d/io/mesh_data.py → src/b3d/io/mesh_data.py
diff --git a/b3d/io/utils.py → src/b3d/io/utils.py b/b3d/io/utils.py → src/b3d/io/utils.py
diff --git a/b3d/io/video_input.py → src/b3d/io/video_input.py b/b3d/io/video_input.py → src/b3d/io/video_input.py
diff --git a/b3d/mesh.py → src/b3d/mesh.py b/b3d/mesh.py → src/b3d/mesh.py
diff --git a/b3d/modeling_utils.py → src/b3d/modeling_utils.py b/b3d/modeling_utils.py → src/b3d/modeling_utils.py
diff --git a/b3d/pose/__init__.py → src/b3d/pose/__init__.py b/b3d/pose/__init__.py → src/b3d/pose/__init__.py
diff --git a/b3d/pose/core.py → src/b3d/pose/core.py b/b3d/pose/core.py → src/b3d/pose/core.py
diff --git a/b3d/pose/pose_utils.py → src/b3d/pose/pose_utils.py b/b3d/pose/pose_utils.py → src/b3d/pose/pose_utils.py
diff --git a/b3d/renderer/LICENSE.txt → src/b3d/renderer/LICENSE.txt b/b3d/renderer/LICENSE.txt → src/b3d/renderer/LICENSE.txt
diff --git a/b3d/renderer/__init__.py → src/b3d/renderer/__init__.py b/b3d/renderer/__init__.py → src/b3d/renderer/__init__.py
diff --git a/b3d/renderer/nvdiffrast/__init__.py → src/b3d/renderer/nvdiffrast/__init__.py b/b3d/renderer/nvdiffrast/__init__.py → src/b3d/renderer/nvdiffrast/__init__.py
diff --git a/b3d/renderer/nvdiffrast/common/__init__.py → ...3d/renderer/nvdiffrast/common/__init__.py b/b3d/renderer/nvdiffrast/common/__init__.py → ...3d/renderer/nvdiffrast/common/__init__.py
diff --git a/b3d/renderer/nvdiffrast/common/common.cpp → ...b3d/renderer/nvdiffrast/common/common.cpp b/b3d/renderer/nvdiffrast/common/common.cpp → ...b3d/renderer/nvdiffrast/common/common.cpp
diff --git a/b3d/renderer/nvdiffrast/common/common.h → src/b3d/renderer/nvdiffrast/common/common.h b/b3d/renderer/nvdiffrast/common/common.h → src/b3d/renderer/nvdiffrast/common/common.h
diff --git a/b3d/renderer/nvdiffrast/common/framework.h → ...3d/renderer/nvdiffrast/common/framework.h b/b3d/renderer/nvdiffrast/common/framework.h → ...3d/renderer/nvdiffrast/common/framework.h
diff --git a/b3d/renderer/nvdiffrast/common/glutil.cpp → ...b3d/renderer/nvdiffrast/common/glutil.cpp b/b3d/renderer/nvdiffrast/common/glutil.cpp → ...b3d/renderer/nvdiffrast/common/glutil.cpp
diff --git a/b3d/renderer/nvdiffrast/common/glutil.h → src/b3d/renderer/nvdiffrast/common/glutil.h b/b3d/renderer/nvdiffrast/common/glutil.h → src/b3d/renderer/nvdiffrast/common/glutil.h
diff --git a/...nderer/nvdiffrast/common/glutil_extlist.h → ...nderer/nvdiffrast/common/glutil_extlist.h b/...nderer/nvdiffrast/common/glutil_extlist.h → ...nderer/nvdiffrast/common/glutil_extlist.h
diff --git a/...renderer/nvdiffrast/common/interpolate.cu → ...renderer/nvdiffrast/common/interpolate.cu b/...renderer/nvdiffrast/common/interpolate.cu → ...renderer/nvdiffrast/common/interpolate.cu
diff --git a/b3d/renderer/nvdiffrast/common/interpolate.h → .../renderer/nvdiffrast/common/interpolate.h b/b3d/renderer/nvdiffrast/common/interpolate.h → .../renderer/nvdiffrast/common/interpolate.h
diff --git a/b3d/renderer/nvdiffrast/common/ops.py → src/b3d/renderer/nvdiffrast/common/ops.py b/b3d/renderer/nvdiffrast/common/ops.py → src/b3d/renderer/nvdiffrast/common/ops.py
diff --git a/b3d/renderer/nvdiffrast/common/rasterize.cu → ...d/renderer/nvdiffrast/common/rasterize.cu b/b3d/renderer/nvdiffrast/common/rasterize.cu → ...d/renderer/nvdiffrast/common/rasterize.cu
diff --git a/b3d/renderer/nvdiffrast/common/rasterize.h → ...3d/renderer/nvdiffrast/common/rasterize.h b/b3d/renderer/nvdiffrast/common/rasterize.h → ...3d/renderer/nvdiffrast/common/rasterize.h
diff --git a/...nderer/nvdiffrast/common/rasterize_gl.cpp → ...nderer/nvdiffrast/common/rasterize_gl.cpp b/...nderer/nvdiffrast/common/rasterize_gl.cpp → ...nderer/nvdiffrast/common/rasterize_gl.cpp
diff --git a/...renderer/nvdiffrast/common/rasterize_gl.h → ...renderer/nvdiffrast/common/rasterize_gl.h b/...renderer/nvdiffrast/common/rasterize_gl.h → ...renderer/nvdiffrast/common/rasterize_gl.h
diff --git a/...nderer/nvdiffrast/common/torch_common.inl → ...nderer/nvdiffrast/common/torch_common.inl b/...nderer/nvdiffrast/common/torch_common.inl → ...nderer/nvdiffrast/common/torch_common.inl
diff --git a/b3d/renderer/nvdiffrast/common/torch_types.h → .../renderer/nvdiffrast/common/torch_types.h b/b3d/renderer/nvdiffrast/common/torch_types.h → .../renderer/nvdiffrast/common/torch_types.h
diff --git a/b3d/renderer/nvdiffrast/jax/__init__.py → src/b3d/renderer/nvdiffrast/jax/__init__.py b/b3d/renderer/nvdiffrast/jax/__init__.py → src/b3d/renderer/nvdiffrast/jax/__init__.py
diff --git a/b3d/renderer/nvdiffrast/jax/bindings.h → src/b3d/renderer/nvdiffrast/jax/bindings.h b/b3d/renderer/nvdiffrast/jax/bindings.h → src/b3d/renderer/nvdiffrast/jax/bindings.h
diff --git a/b3d/renderer/nvdiffrast/jax/jax_bindings.cpp → .../renderer/nvdiffrast/jax/jax_bindings.cpp b/b3d/renderer/nvdiffrast/jax/jax_bindings.cpp → .../renderer/nvdiffrast/jax/jax_bindings.cpp
diff --git a/b3d/renderer/nvdiffrast/jax/jax_bindings.h → ...3d/renderer/nvdiffrast/jax/jax_bindings.h b/b3d/renderer/nvdiffrast/jax/jax_bindings.h → ...3d/renderer/nvdiffrast/jax/jax_bindings.h
diff --git a/...nderer/nvdiffrast/jax/jax_interpolate.cpp → ...nderer/nvdiffrast/jax/jax_interpolate.cpp b/...nderer/nvdiffrast/jax/jax_interpolate.cpp → ...nderer/nvdiffrast/jax/jax_interpolate.cpp
diff --git a/...renderer/nvdiffrast/jax/jax_interpolate.h → ...renderer/nvdiffrast/jax/jax_interpolate.h b/...renderer/nvdiffrast/jax/jax_interpolate.h → ...renderer/nvdiffrast/jax/jax_interpolate.h
diff --git a/...derer/nvdiffrast/jax/jax_rasterize_gl.cpp → ...derer/nvdiffrast/jax/jax_rasterize_gl.cpp b/...derer/nvdiffrast/jax/jax_rasterize_gl.cpp → ...derer/nvdiffrast/jax/jax_rasterize_gl.cpp
diff --git a/...enderer/nvdiffrast/jax/jax_rasterize_gl.h → ...enderer/nvdiffrast/jax/jax_rasterize_gl.h b/...enderer/nvdiffrast/jax/jax_rasterize_gl.h → ...enderer/nvdiffrast/jax/jax_rasterize_gl.h
diff --git a/b3d/renderer/nvdiffrast/jax/ops.py → src/b3d/renderer/nvdiffrast/jax/ops.py b/b3d/renderer/nvdiffrast/jax/ops.py → src/b3d/renderer/nvdiffrast/jax/ops.py
diff --git a/b3d/renderer/nvdiffrast/lib/setgpu.lib → src/b3d/renderer/nvdiffrast/lib/setgpu.lib b/b3d/renderer/nvdiffrast/lib/setgpu.lib → src/b3d/renderer/nvdiffrast/lib/setgpu.lib
diff --git a/b3d/renderer/nvdiffrast_jax/LICENSE.txt → src/b3d/renderer/nvdiffrast_jax/LICENSE.txt b/b3d/renderer/nvdiffrast_jax/LICENSE.txt → src/b3d/renderer/nvdiffrast_jax/LICENSE.txt
diff --git a/b3d/renderer/nvdiffrast_jax/README.md → src/b3d/renderer/nvdiffrast_jax/README.md b/b3d/renderer/nvdiffrast_jax/README.md → src/b3d/renderer/nvdiffrast_jax/README.md
diff --git a/b3d/renderer/nvdiffrast_jax/__init__.py → src/b3d/renderer/nvdiffrast_jax/__init__.py b/b3d/renderer/nvdiffrast_jax/__init__.py → src/b3d/renderer/nvdiffrast_jax/__init__.py
diff --git a/...iffrast_jax/nvdiffrast/common/__init__.py → ...iffrast_jax/nvdiffrast/common/__init__.py b/...iffrast_jax/nvdiffrast/common/__init__.py → ...iffrast_jax/nvdiffrast/common/__init__.py
diff --git a/...diffrast_jax/nvdiffrast/common/common.cpp → ...diffrast_jax/nvdiffrast/common/common.cpp b/...diffrast_jax/nvdiffrast/common/common.cpp → ...diffrast_jax/nvdiffrast/common/common.cpp
diff --git a/...nvdiffrast_jax/nvdiffrast/common/common.h → ...nvdiffrast_jax/nvdiffrast/common/common.h b/...nvdiffrast_jax/nvdiffrast/common/common.h → ...nvdiffrast_jax/nvdiffrast/common/common.h
diff --git a/...iffrast_jax/nvdiffrast/common/framework.h → ...iffrast_jax/nvdiffrast/common/framework.h b/...iffrast_jax/nvdiffrast/common/framework.h → ...iffrast_jax/nvdiffrast/common/framework.h
diff --git a/...diffrast_jax/nvdiffrast/common/glutil.cpp → ...diffrast_jax/nvdiffrast/common/glutil.cpp b/...diffrast_jax/nvdiffrast/common/glutil.cpp → ...diffrast_jax/nvdiffrast/common/glutil.cpp
diff --git a/...nvdiffrast_jax/nvdiffrast/common/glutil.h → ...nvdiffrast_jax/nvdiffrast/common/glutil.h b/...nvdiffrast_jax/nvdiffrast/common/glutil.h → ...nvdiffrast_jax/nvdiffrast/common/glutil.h
diff --git a/...st_jax/nvdiffrast/common/glutil_extlist.h → ...st_jax/nvdiffrast/common/glutil_extlist.h b/...st_jax/nvdiffrast/common/glutil_extlist.h → ...st_jax/nvdiffrast/common/glutil_extlist.h
diff --git a/...rast_jax/nvdiffrast/common/interpolate.cu → ...rast_jax/nvdiffrast/common/interpolate.cu b/...rast_jax/nvdiffrast/common/interpolate.cu → ...rast_jax/nvdiffrast/common/interpolate.cu
diff --git a/...frast_jax/nvdiffrast/common/interpolate.h → ...frast_jax/nvdiffrast/common/interpolate.h b/...frast_jax/nvdiffrast/common/interpolate.h → ...frast_jax/nvdiffrast/common/interpolate.h
diff --git a/...r/nvdiffrast_jax/nvdiffrast/common/ops.py → ...r/nvdiffrast_jax/nvdiffrast/common/ops.py b/...r/nvdiffrast_jax/nvdiffrast/common/ops.py → ...r/nvdiffrast_jax/nvdiffrast/common/ops.py
diff --git a/...ffrast_jax/nvdiffrast/common/rasterize.cu → ...ffrast_jax/nvdiffrast/common/rasterize.cu b/...ffrast_jax/nvdiffrast/common/rasterize.cu → ...ffrast_jax/nvdiffrast/common/rasterize.cu
diff --git a/...iffrast_jax/nvdiffrast/common/rasterize.h → ...iffrast_jax/nvdiffrast/common/rasterize.h b/...iffrast_jax/nvdiffrast/common/rasterize.h → ...iffrast_jax/nvdiffrast/common/rasterize.h
diff --git a/...st_jax/nvdiffrast/common/rasterize_gl.cpp → ...st_jax/nvdiffrast/common/rasterize_gl.cpp b/...st_jax/nvdiffrast/common/rasterize_gl.cpp → ...st_jax/nvdiffrast/common/rasterize_gl.cpp
diff --git a/...rast_jax/nvdiffrast/common/rasterize_gl.h → ...rast_jax/nvdiffrast/common/rasterize_gl.h b/...rast_jax/nvdiffrast/common/rasterize_gl.h → ...rast_jax/nvdiffrast/common/rasterize_gl.h
diff --git a/...st_jax/nvdiffrast/common/torch_common.inl → ...st_jax/nvdiffrast/common/torch_common.inl b/...st_jax/nvdiffrast/common/torch_common.inl → ...st_jax/nvdiffrast/common/torch_common.inl
diff --git a/...frast_jax/nvdiffrast/common/torch_types.h → ...frast_jax/nvdiffrast/common/torch_types.h b/...frast_jax/nvdiffrast/common/torch_types.h → ...frast_jax/nvdiffrast/common/torch_types.h
diff --git a/...nvdiffrast_jax/nvdiffrast/jax/__init__.py → ...nvdiffrast_jax/nvdiffrast/jax/__init__.py b/...nvdiffrast_jax/nvdiffrast/jax/__init__.py → ...nvdiffrast_jax/nvdiffrast/jax/__init__.py
diff --git a/...rast_jax/nvdiffrast/jax/jax_binding_ops.h → ...rast_jax/nvdiffrast/jax/jax_binding_ops.h b/...rast_jax/nvdiffrast/jax/jax_binding_ops.h → ...rast_jax/nvdiffrast/jax/jax_binding_ops.h
diff --git a/...frast_jax/nvdiffrast/jax/jax_bindings.cpp → ...frast_jax/nvdiffrast/jax/jax_bindings.cpp b/...frast_jax/nvdiffrast/jax/jax_bindings.cpp → ...frast_jax/nvdiffrast/jax/jax_bindings.cpp
diff --git a/...st_jax/nvdiffrast/jax/jax_interpolate.cpp → ...st_jax/nvdiffrast/jax/jax_interpolate.cpp b/...st_jax/nvdiffrast/jax/jax_interpolate.cpp → ...st_jax/nvdiffrast/jax/jax_interpolate.cpp
diff --git a/...rast_jax/nvdiffrast/jax/jax_interpolate.h → ...rast_jax/nvdiffrast/jax/jax_interpolate.h b/...rast_jax/nvdiffrast/jax/jax_interpolate.h → ...rast_jax/nvdiffrast/jax/jax_interpolate.h
diff --git a/...t_jax/nvdiffrast/jax/jax_rasterize_gl.cpp → ...t_jax/nvdiffrast/jax/jax_rasterize_gl.cpp b/...t_jax/nvdiffrast/jax/jax_rasterize_gl.cpp → ...t_jax/nvdiffrast/jax/jax_rasterize_gl.cpp
diff --git a/...ast_jax/nvdiffrast/jax/jax_rasterize_gl.h → ...ast_jax/nvdiffrast/jax/jax_rasterize_gl.h b/...ast_jax/nvdiffrast/jax/jax_rasterize_gl.h → ...ast_jax/nvdiffrast/jax/jax_rasterize_gl.h
diff --git a/...erer/nvdiffrast_jax/nvdiffrast/jax/ops.py → ...erer/nvdiffrast_jax/nvdiffrast/jax/ops.py b/...erer/nvdiffrast_jax/nvdiffrast/jax/ops.py → ...erer/nvdiffrast_jax/nvdiffrast/jax/ops.py
diff --git a/...frast_jax/nvdiffrast/jax/torch_common.inl → ...frast_jax/nvdiffrast/jax/torch_common.inl b/...frast_jax/nvdiffrast/jax/torch_common.inl → ...frast_jax/nvdiffrast/jax/torch_common.inl
diff --git a/...diffrast_jax/nvdiffrast/jax/torch_types.h → ...diffrast_jax/nvdiffrast/jax/torch_types.h b/...diffrast_jax/nvdiffrast/jax/torch_types.h → ...diffrast_jax/nvdiffrast/jax/torch_types.h
diff --git a/.../nvdiffrast_jax/nvdiffrast/lib/setgpu.lib → .../nvdiffrast_jax/nvdiffrast/lib/setgpu.lib b/.../nvdiffrast_jax/nvdiffrast/lib/setgpu.lib → .../nvdiffrast_jax/nvdiffrast/lib/setgpu.lib
diff --git a/...diffrast_jax/renderer_matching_pytorch.py → ...diffrast_jax/renderer_matching_pytorch.py b/...diffrast_jax/renderer_matching_pytorch.py → ...diffrast_jax/renderer_matching_pytorch.py
diff --git a/b3d/renderer/nvdiffrast_jax/run_sample.sh → ...b3d/renderer/nvdiffrast_jax/run_sample.sh b/b3d/renderer/nvdiffrast_jax/run_sample.sh → ...b3d/renderer/nvdiffrast_jax/run_sample.sh
diff --git a/b3d/renderer/nvdiffrast_jax/scripts/test.sh → ...d/renderer/nvdiffrast_jax/scripts/test.sh b/b3d/renderer/nvdiffrast_jax/scripts/test.sh → ...d/renderer/nvdiffrast_jax/scripts/test.sh
diff --git a/b3d/renderer/nvdiffrast_jax/setup.py → src/b3d/renderer/nvdiffrast_jax/setup.py b/b3d/renderer/nvdiffrast_jax/setup.py → src/b3d/renderer/nvdiffrast_jax/setup.py
diff --git a/b3d/renderer/renderer.py → src/b3d/renderer/renderer.py b/b3d/renderer/renderer.py → src/b3d/renderer/renderer.py
diff --git a/b3d/renderer/renderer_original.py → src/b3d/renderer/renderer_original.py b/b3d/renderer/renderer_original.py → src/b3d/renderer/renderer_original.py
diff --git a/b3d/renderer/torch/__init__.py → src/b3d/renderer/torch/__init__.py b/b3d/renderer/torch/__init__.py → src/b3d/renderer/torch/__init__.py
diff --git a/b3d/renderer/torch/pose.py → src/b3d/renderer/torch/pose.py b/b3d/renderer/torch/pose.py → src/b3d/renderer/torch/pose.py
diff --git a/b3d/renderer/torch/renderutils/__init__.py → ...3d/renderer/torch/renderutils/__init__.py b/b3d/renderer/torch/renderutils/__init__.py → ...3d/renderer/torch/renderutils/__init__.py
diff --git a/...nderer/torch/renderutils/c_src/common.cpp → ...nderer/torch/renderutils/c_src/common.cpp b/...nderer/torch/renderutils/c_src/common.cpp → ...nderer/torch/renderutils/c_src/common.cpp
diff --git a/...renderer/torch/renderutils/c_src/common.h → ...renderer/torch/renderutils/c_src/common.h b/...renderer/torch/renderutils/c_src/common.h → ...renderer/torch/renderutils/c_src/common.h
diff --git a/b3d/renderer/torch/renderutils/c_src/mesh.cu → .../renderer/torch/renderutils/c_src/mesh.cu b/b3d/renderer/torch/renderutils/c_src/mesh.cu → .../renderer/torch/renderutils/c_src/mesh.cu
diff --git a/b3d/renderer/torch/renderutils/c_src/mesh.h → ...d/renderer/torch/renderutils/c_src/mesh.h b/b3d/renderer/torch/renderutils/c_src/mesh.h → ...d/renderer/torch/renderutils/c_src/mesh.h
diff --git a/...renderer/torch/renderutils/c_src/tensor.h → ...renderer/torch/renderutils/c_src/tensor.h b/...renderer/torch/renderutils/c_src/tensor.h → ...renderer/torch/renderutils/c_src/tensor.h
diff --git a/...orch/renderutils/c_src/torch_bindings.cpp → ...orch/renderutils/c_src/torch_bindings.cpp b/...orch/renderutils/c_src/torch_bindings.cpp → ...orch/renderutils/c_src/torch_bindings.cpp
diff --git a/b3d/renderer/torch/renderutils/c_src/vec3f.h → .../renderer/torch/renderutils/c_src/vec3f.h b/b3d/renderer/torch/renderutils/c_src/vec3f.h → .../renderer/torch/renderutils/c_src/vec3f.h
diff --git a/b3d/renderer/torch/renderutils/c_src/vec4f.h → .../renderer/torch/renderutils/c_src/vec4f.h b/b3d/renderer/torch/renderutils/c_src/vec4f.h → .../renderer/torch/renderutils/c_src/vec4f.h
diff --git a/b3d/renderer/torch/renderutils/diffdope.py → ...3d/renderer/torch/renderutils/diffdope.py b/b3d/renderer/torch/renderutils/diffdope.py → ...3d/renderer/torch/renderutils/diffdope.py
diff --git a/b3d/renderer/torch/renderutils/ops.py → src/b3d/renderer/torch/renderutils/ops.py b/b3d/renderer/torch/renderutils/ops.py → src/b3d/renderer/torch/renderutils/ops.py
diff --git a/b3d/renderer/torch/torch_utils.py → src/b3d/renderer/torch/torch_utils.py b/b3d/renderer/torch/torch_utils.py → src/b3d/renderer/torch/torch_utils.py
diff --git a/b3d/types.py → src/b3d/types.py b/b3d/types.py → src/b3d/types.py
diff --git a/b3d/utils.py → src/b3d/utils.py b/b3d/utils.py → src/b3d/utils.py
@@ -29,7 +29,7 @@
 
 
 def get_root_path() -> Path:
-    return Path(Path(b3d.__file__).parents[1])
+    return Path(Path(b3d.__file__).parents[2])
 
 
 def get_assets() -> Path: