NVIDIA-Merlin/dataloader

The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX

PythonShelldeep-learningtensorflowpytorchrecommender-systemsjax
This is stars and forks stats for /NVIDIA-Merlin/dataloader repository. As of 29 Apr, 2024 this repository has 391 stars and 22 forks.

Merlin Dataloader The merlin-dataloader lets you quickly train recommender models for TensorFlow, PyTorch and JAX. It eliminates the biggest bottleneck in training recommender models, by providing GPU optimized dataloaders that read data directly into the GPU, and then do a 0-copy transfer to TensorFlow and PyTorch using dlpack. The benefits of the Merlin Dataloader include: Over 10x speedup over native framework dataloaders Handles larger than memory datasets Per-epoch shuffling Distributed training Installation Merlin-dataloader...
Read on GithubGithub Stats Page
repotechsstarsweeklyforksweekly
comet-ml/kangasJupyter NotebookPythonJavaScript9890430
estiens/world_cup_jsonRubyHamlHTML93001090
rustodon/rustodonRustHTMLCSS831044+1
ohjelmistotuotanto-hy/ohjelmistotuotanto-hy.github.ioSCSSHTMLRuby50850
gloeyisk/universal-gms-dozeShell1k0250
openshift-helm-charts/stageSmartyMakefilePython2090
LeastAuthority/moonmath-manualTeXShell2520460
riscv-non-isa/riscv-rvm-csiCPythonMakefile8050
bcgov/PSPTSQLC#TypeScript70240
Oneplus/dotfilesVim ScriptPythonR0020