NVIDIA-Merlin/dataloader
GPU-optimized dataloader for training deep learning recommender models with TensorFlow, PyTorch, and JAX frameworks

Velocity · 7d
+0.3
★ / day
Trend
→steady
star history
NVIDIA Merlin Dataloader accelerates training of recommender models by loading data directly into GPU memory and providing zero-copy transfers to deep learning frameworks via dlpack. It claims over 10x speedup over native framework dataloaders, handles datasets larger than memory, provides per-epoch shuffling, and supports distributed training across TensorFlow, PyTorch, and JAX.