#96 · Primary category: Deep Learning Frameworks
tiny-cuda-nn
Lightning fast C++/CUDA neural network framework
Project last updated:04/21/26
GitHub Stars
4.5K
Forks
577
Contributors
36
License
Other
Why we included this project
tiny-cuda-nn is the framework behind NVIDIA's Instant NGP, and that alone tells you what it is for: a compact C++/CUDA library that trains and queries small neural networks fast enough for real-time work. Its two signature pieces are a fully fused multilayer perceptron and a multiresolution hash encoding, each covered by its own technical paper, and together they let a network run as a single fused kernel instead of stepping back into TensorFlow or PyTorch layer by layer. The README benchmarks make the payoff concrete, measuring against TensorFlow 2.5 with XLA on an RTX 3090, and the gap is large. Model setup happens through JSON, so swapping an encoding, loss, or optimizer does not require touching code, and PyTorch bindings keep it usable from Python. Rendering and 3D reconstruction are the obvious home, where a trained model gets queried millions of times per frame, but the same fused-kernel speedup applies anywhere you push small, dense networks hard.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.