← all repositories

asappresearch/sru

SRU is a recurrent neural network unit implementation that runs 10-16x faster than LSTM while maintaining accuracy, built for PyTorch.

2.1k stars Python ML Frameworks
sru
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

SRU (Simple Recurrent Units) is a high-performance recurrent unit library for PyTorch that achieves significant speed-ups over standard LSTM by enabling highly parallelizable computation. The library implements both the original SRU architecture from EMNLP 2018 and the newer SRU++ variant from 2021, which combines attention mechanisms with fast recurrence. It supports installation via pip or source and requires PyTorch and ninja build system.

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