Switch language한국어
Back to the list

Block-sparse GPU kernels

TL;DR AI

Key summary

2 min read
  1. Researchers released optimized GPU kernels for block-sparse neural networks, speeding up sparse linear ops in fully connected and convolutional layers.

  2. The kernels support TensorFlow and can work with models like LSTMs, aiming to make sparsity practical on GPUs.

  3. They report state-of-the-art results on some sentiment analysis and generative modeling tasks, including runs on an NVIDIA Titan X Pascal.

  4. By sharply reducing compute and memory needs, the approach could enable much larger models and new sparsity-based architectures.

Read the original