A PyTorch Library for Accelerating 3D Deep Learning Research
Python
5,184
414 commits
updated Oct 7, 2026
TL;DR — Kaolin is NVIDIA's PyTorch library of reusable, GPU-optimized modules distilled from 3D deep learning research.
- Representation-agnostic physics (meshes, splats, point clouds) with Simplicits
- Differentiable rendering (DIB-R, nvdiffrast, easy_render PBR)
- First-class 3D Gaussian splats (PLY/USD I/O, densification, gsplat bridge)
- GPU octree acceleration structure: Structured Point Clouds (SPC)
- Conversions between 3D representations, quaternion ops, and USD I/O
Kaolin packages reusable building blocks from NVIDIA 3D research into a cohesive PyTorch API — continuously improving representation-agnostic physics simulation, fast conversions between representations, quaternion math, batched mesh and splat containers, I/O, visualization and more. See kaolin.readthedocs.io for tutorials and API reference, and developer.nvidia.com/kaolin for the NVIDIA Kaolin hub.
Join us in Los Angeles! Two sessions showcasing new Kaolin capabilities. Full details here.
Talk — Any Representation, Any Hardware, All Interactions Accelerating Interactive Prototypes Over Cutting Edge AI & 3D Research Sun 19 Jul 2026, 3:00–3:20 pm PDT · Room 408 A Maria Shugrina (NVIDIA / University of Toronto) Introduces Kaolin's new web client-server framework ( |
Hands-on Lab — From 3D Captures to Simulated Digital Environments Thu 23 Jul 2026, 10:15–11:45 am PDT · Concourse Hall Clement Fuji Tsang, Vismay Modi, Maria Shugrina (NVIDIA) A capture-to-simulation pipeline for in-the-wild 3D Gaussian Splat scenes: segment objects, predict volumetric mechanical properties, run mixed splat–mesh physics — powered by recent Kaolin features. Export shareable USD files with Kaolin's custom physics schema. Bring your laptop and follow along hands-on. Full Conference badge required. |
|
Physics (Simplicits) Simulate meshes, splats, and point clouds with collisions. Warp-accelerated, representation-agnostic. |
3D Gaussian Splats
|
Differentiable Rendering DIB-R, nvdiffrast, Docs · DIB-R · Easy render · Camera · Lighting |
|
Structured Point Clouds GPU octree acceleration structure with ray tracing and feature grids. |
USD Pipeline Import/export meshes, point clouds, gaussians, and physics materials with Kaolin's custom schema. API · Checkpoints · GLTF viz |
Visualization Jupyter 3D viewer, Timelapse checkpoints, and web client-server framework. |
|
Conversions Fast GPU conversions between meshes, voxel grids, point clouds, gaussians, and more. Docs · DMTet · FlexiCubes |
Quaternions Differentiable quaternion and rigid-transform utilities for 3D deep learning. |
Surface Meshes Batched Docs · Tutorial · Easy render |
Experimental: Newton coupling — Simplicits soft bodies with rigid bodies, MPM, and articulated robots (rigid · MPM · Franka).
See the tutorial index and API reference at kaolin.readthedocs.io.
Starting with v0.12.0, Kaolin supports installation with pre-built wheels:
# Replace TORCH_VERSION and CUDA_VERSION with your torch / cuda versions
pip install kaolin==0.18.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-{TORCH_VERSION}_cu{CUDA_VERSION}.html
For example, kaolin 0.18.0 with PyTorch 2.8.0 and CUDA 12.8:
pip install kaolin==0.18.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.8.0_cu128.html
See the installation guide for the full torch/CUDA compatibility matrix and source install instructions.
python -c "import kaolin; print(kaolin.__version__)"
import kaolin
print(kaolin.__version__)
Load a 3D Gaussian splat from PLY or USD:
import kaolin
from kaolin.rep import GaussianSplatModel
gs = kaolin.io.import_gaussiancloud("scene.ply")
print(gs) # GaussianSplatModel with positions, scales, rotations, opacities, ...
Simulate a mesh with Simplicits — see the physics tutorial.
Render a mesh with the easy PBR API:
import kaolin as kal
mesh = kal.io.obj.import_mesh("model.obj")
camera = kal.render.easy_render.default_camera(512)
lighting = kal.render.easy_render.default_lighting()
result = kal.render.easy_render.render_mesh(camera, mesh, lighting=lighting)
Recent work on master since v0.18.0:
kaolin/visualize/dash) — rapid prototyping of interactive Web UIs over AI and 3D research (web_framework_prerelease branch; SIGGRAPH 2026 talk)GaussianSplatModel and PointSamples tensor-container APIkaolin/ops/conversions/flexicubes/See release notes for details.
Notebooks live under examples/tutorial/. Highlights by topic:
Physics
Gaussians
Rendering
Structured Point Clouds
Visualization
Full index: tutorial index.
Projects built with Kaolin:
Please review our contribution guidelines.
Kaolin is released under the Apache License 2.0. A default import kaolin gives you the full Apache-licensed library.
The kaolin/non_commercial/ package is legacy only — kept for backward compatibility with older import paths (e.g. the pre-Apache FlexiCubes copy). New code should use the Apache-licensed modules under kaolin/ops/, kaolin/rep/, and the rest of the package tree.
If you use Kaolin in your research, please cite:
@software{KaolinLibrary,
author = {Tsang, Clement Fuji and Shugrina, Maria and Lafleche, Jean-Francois and Perel, Or and Loop, Charles and Takikawa, Towaki and Modi, Vismay and Zook, Alexander and Wang, Jiehan and Chen, Wenzheng and Shen, Tianchang and Gao, Jun and Jatavallabhula, Krishna Murthy and Smith, Edward and Rozantsev, Artem and Fidler, Sanja and State, Gavriel and Gorski, Jason and Xiang, Tommy and Li, Jianing and Li, Michael and Lebaredian, Rev},
title = {{Kaolin: A PyTorch Library for Accelerating 3D Deep Learning Research}},
version = {0.18.0},
date = {2024-11-20},
url = {https://github.com/NVIDIAGameWorks/kaolin}
}
Current team: Clement Fuji Tsang (Technical Lead), Maria (Masha) Shugrina (Manager), Charles Loop, Vismay Modi, Or Perel
Other major contributors: Alexander Zook, Donglai Xiang, Wenzheng Chen, Sanja Fidler, Jun Gao, Jason Gorski, Jean-Francois Lafleche, Rev Lebaredian, Jianing Li, Michael Li, Krishna Murthy Jatavallabhula, Artem Rozantsev, Tianchang (Frank) Shen, Edward Smith, Gavriel State, Towaki Takikawa, Jiehan Wang, Tommy Xiang
310 followers · starred Oct 2022
221 followers · starred Jul 2023
265 followers · starred Nov 2019
200 followers · starred Dec 2021
A PyTorch Library for Accelerating 3D Deep Learning Research
Python
5,184
414 commits
updated Oct 7, 2026
TL;DR — Kaolin is NVIDIA's PyTorch library of reusable, GPU-optimized modules distilled from 3D deep learning research.
- Representation-agnostic physics (meshes, splats, point clouds) with Simplicits
- Differentiable rendering (DIB-R, nvdiffrast, easy_render PBR)
- First-class 3D Gaussian splats (PLY/USD I/O, densification, gsplat bridge)
- GPU octree acceleration structure: Structured Point Clouds (SPC)
- Conversions between 3D representations, quaternion ops, and USD I/O
Kaolin packages reusable building blocks from NVIDIA 3D research into a cohesive PyTorch API — continuously improving representation-agnostic physics simulation, fast conversions between representations, quaternion math, batched mesh and splat containers, I/O, visualization and more. See kaolin.readthedocs.io for tutorials and API reference, and developer.nvidia.com/kaolin for the NVIDIA Kaolin hub.
Join us in Los Angeles! Two sessions showcasing new Kaolin capabilities. Full details here.
Talk — Any Representation, Any Hardware, All Interactions Accelerating Interactive Prototypes Over Cutting Edge AI & 3D Research Sun 19 Jul 2026, 3:00–3:20 pm PDT · Room 408 A Maria Shugrina (NVIDIA / University of Toronto) Introduces Kaolin's new web client-server framework ( |
Hands-on Lab — From 3D Captures to Simulated Digital Environments Thu 23 Jul 2026, 10:15–11:45 am PDT · Concourse Hall Clement Fuji Tsang, Vismay Modi, Maria Shugrina (NVIDIA) A capture-to-simulation pipeline for in-the-wild 3D Gaussian Splat scenes: segment objects, predict volumetric mechanical properties, run mixed splat–mesh physics — powered by recent Kaolin features. Export shareable USD files with Kaolin's custom physics schema. Bring your laptop and follow along hands-on. Full Conference badge required. |
|
Physics (Simplicits) Simulate meshes, splats, and point clouds with collisions. Warp-accelerated, representation-agnostic. |
3D Gaussian Splats
|
Differentiable Rendering DIB-R, nvdiffrast, Docs · DIB-R · Easy render · Camera · Lighting |
|
Structured Point Clouds GPU octree acceleration structure with ray tracing and feature grids. |
USD Pipeline Import/export meshes, point clouds, gaussians, and physics materials with Kaolin's custom schema. API · Checkpoints · GLTF viz |
Visualization Jupyter 3D viewer, Timelapse checkpoints, and web client-server framework. |
|
Conversions Fast GPU conversions between meshes, voxel grids, point clouds, gaussians, and more. Docs · DMTet · FlexiCubes |
Quaternions Differentiable quaternion and rigid-transform utilities for 3D deep learning. |
Surface Meshes Batched Docs · Tutorial · Easy render |
Experimental: Newton coupling — Simplicits soft bodies with rigid bodies, MPM, and articulated robots (rigid · MPM · Franka).
See the tutorial index and API reference at kaolin.readthedocs.io.
Starting with v0.12.0, Kaolin supports installation with pre-built wheels:
# Replace TORCH_VERSION and CUDA_VERSION with your torch / cuda versions
pip install kaolin==0.18.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-{TORCH_VERSION}_cu{CUDA_VERSION}.html
For example, kaolin 0.18.0 with PyTorch 2.8.0 and CUDA 12.8:
pip install kaolin==0.18.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.8.0_cu128.html
See the installation guide for the full torch/CUDA compatibility matrix and source install instructions.
python -c "import kaolin; print(kaolin.__version__)"
import kaolin
print(kaolin.__version__)
Load a 3D Gaussian splat from PLY or USD:
import kaolin
from kaolin.rep import GaussianSplatModel
gs = kaolin.io.import_gaussiancloud("scene.ply")
print(gs) # GaussianSplatModel with positions, scales, rotations, opacities, ...
Simulate a mesh with Simplicits — see the physics tutorial.
Render a mesh with the easy PBR API:
import kaolin as kal
mesh = kal.io.obj.import_mesh("model.obj")
camera = kal.render.easy_render.default_camera(512)
lighting = kal.render.easy_render.default_lighting()
result = kal.render.easy_render.render_mesh(camera, mesh, lighting=lighting)
Recent work on master since v0.18.0:
kaolin/visualize/dash) — rapid prototyping of interactive Web UIs over AI and 3D research (web_framework_prerelease branch; SIGGRAPH 2026 talk)GaussianSplatModel and PointSamples tensor-container APIkaolin/ops/conversions/flexicubes/See release notes for details.
Notebooks live under examples/tutorial/. Highlights by topic:
Physics
Gaussians
Rendering
Structured Point Clouds
Visualization
Full index: tutorial index.
Projects built with Kaolin:
Please review our contribution guidelines.
Kaolin is released under the Apache License 2.0. A default import kaolin gives you the full Apache-licensed library.
The kaolin/non_commercial/ package is legacy only — kept for backward compatibility with older import paths (e.g. the pre-Apache FlexiCubes copy). New code should use the Apache-licensed modules under kaolin/ops/, kaolin/rep/, and the rest of the package tree.
If you use Kaolin in your research, please cite:
@software{KaolinLibrary,
author = {Tsang, Clement Fuji and Shugrina, Maria and Lafleche, Jean-Francois and Perel, Or and Loop, Charles and Takikawa, Towaki and Modi, Vismay and Zook, Alexander and Wang, Jiehan and Chen, Wenzheng and Shen, Tianchang and Gao, Jun and Jatavallabhula, Krishna Murthy and Smith, Edward and Rozantsev, Artem and Fidler, Sanja and State, Gavriel and Gorski, Jason and Xiang, Tommy and Li, Jianing and Li, Michael and Lebaredian, Rev},
title = {{Kaolin: A PyTorch Library for Accelerating 3D Deep Learning Research}},
version = {0.18.0},
date = {2024-11-20},
url = {https://github.com/NVIDIAGameWorks/kaolin}
}
Current team: Clement Fuji Tsang (Technical Lead), Maria (Masha) Shugrina (Manager), Charles Loop, Vismay Modi, Or Perel
Other major contributors: Alexander Zook, Donglai Xiang, Wenzheng Chen, Sanja Fidler, Jun Gao, Jason Gorski, Jean-Francois Lafleche, Rev Lebaredian, Jianing Li, Michael Li, Krishna Murthy Jatavallabhula, Artem Rozantsev, Tianchang (Frank) Shen, Edward Smith, Gavriel State, Towaki Takikawa, Jiehan Wang, Tommy Xiang
310 followers · starred Oct 2022
221 followers · starred Jul 2023
265 followers · starred Nov 2019
200 followers · starred Dec 2021