← all repositories

NVlabs/tiny-cuda-nn

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

tiny-cuda-nn
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.