Accelerating fuzzy document deduplication to improve LLM training w/ RAPIDS & Dask
summary
NVIDIA NeMo Data Curator uses RAPIDS and Dask to scale fuzzy deduplication of terabytes of text by bucketing with MinHash, expanding pairs into a Dask dataframe, and computing Jaccard similarity on GPU‑accelerated partitions. The workflow runs on a RAPIDS + Dask cluster deployed on Kubernetes and reports dramatic speedups for a 4.5 TB dataset, for example 4 nodes with GPUs finish in 3 hours vs 20 CPU nodes in 37 hours, showing RAPIDS + Dask on Kubernetes cuts time and cost.
AI-written summary · about
chapters (8)
- 0:00Introduction
- 2:48LLM data preprocessing challenges
- 5:13RAPIDS GPU acceleration tools
- 7:23Scaling with Dask
- 11:18Cloud deployment
- 15:30Fuzzy document deduplication workflow
- 21:06Performance gains
- 22:49Q&A
related talks
01h 22mHigh-Performance Data Science at Scale with RAPIDS, Dask, and GPUs
38mRAPIDS- Open Source GPU Data Science
39mAccelerating Data Science with RAPIDS
01h 08mGPU Data Science with Rapids
41mAccelerating Data Science with RAPIDS
34mRAPIDS: How to accelerate your data science pipeline by orders of magnitude
01h 36mRAPIDS: GPU-Accelerated Data Science for PyData Users
20mEnd-to-End Data Science on GPUs
