Jiabao Lei

RGBD-Diffusion. RGBD2: Generative Scene Synthesis via Incremental View Inpainting using RGBD Diffusion Models

armesh. [NeurIPS 2025] ARMesh: Autoregressive Mesh Generation via Next-Level-of-Detail Prediction

segmentator. Segmentator for clustering on meshes or pointclouds

torch_kdtree. A CUDA implementation of KDTree in PyTorch

pytorch_grouped_gemm. High Performance Grouped GEMM in PyTorch

recon. an universal pytorch deep learning experiment codebase

content-aware-fill. Content Aware Fill for Linux's Python

torch_indexing. Auxiliary library for advanced pytorch tensor indexing

instant_nerf. this repository tries to add more function supports to instant-ngp in a pure CUDA C++ level

pytorch_hash. a pytorch library for fast hashing using FNV-1A

pytorch-cuda-cpp-extension-tutorials. 基于PyTorch框架的CUDA/C++扩展算子编写指南（施工中）

geok. computational geometrical kernels

pytorch_searchtree. an efficient nearest/farthest point searching implementation using sparse tree partitioning

indexed_ordered_map. ordered map with optimized indexing access based on order-statistic tree