uber-research/sbnet
A TensorFlow library of custom CUDA operations that implements Sparse Blocks Networks for efficient neural network inference.

Velocity · 7d
+0.1
★ / day
Trend
→steady
star history
This repository contains a TensorFlow custom operations library implementing SBNet (Sparse Blocks Network) for faster inference, along with a Python implementation of sparse ResNet blocks. It provides custom CUDA kernels for ops like reduce_mask, sparse_gather, and sparse_scatter that work on active block regions rather than dense tensors. The library also benchmarks against Submanifold Sparse Convolutional Networks.