NVlabs/tiny-cuda-nn
A high-performance C++/CUDA framework for training and querying neural networks with GPU acceleration.

Velocity · 7d
+2.4
★ / day
Trend
→steady
star history
Tiny CUDA Neural Networks is a lightweight framework for training and querying neural networks on GPUs. It provides fully-fused multi-layer perceptrons and multiresolution hash encodings optimized for speed, supporting various input encodings, loss functions like L2, and optimizers like Adam. The framework exposes a C++/CUDA API and offers optional JIT fusion for additional performance gains.