← all repositories

NVIDIA-Merlin/dataloader

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

423 stars Python Data Tooling
dataloader
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.

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