amazon-archives/amazon-dsstne
Amazon's open-source deep learning library for training and deploying sparse recommendation models across multiple GPUs.

Velocity · 7d
+1.2
★ / day
Trend
→steady
star history
DSSTNE (Deep Scalable Sparse Tensor Network Engine) is a C++ library for building and deploying deep learning recommendation models with sparse inputs and outputs. It supports model-parallel scaling across multiple GPUs, enabling training of networks too large for a single GPU. The library includes custom GPU kernels optimized for sparse computation and was developed for production-scale personalized recommendations at Amazon.