NVIDIAGameWorks / kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).
Other
1.45k stars 132 forks source link

Added typing to functions, and blas construction separated from grid #154

Closed orperel closed 1 year ago

orperel commented 1 year ago

Adds typing to various constructors & functions, as a perquisite for the new upcoming config system. This MR also handles the blas construction outside of the feature grids, which allows for the removal of various grid constructors (this simplifies the grids and makes the upcoming config less convoluted).