← all repositories

uber-research/sbnet

A TensorFlow library of custom CUDA operations that implements Sparse Blocks Networks for efficient neural network inference.

sbnet
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.

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