NVIDIA Ingest is a set of microservices for parsing hundreds of thousands of complex, messy unstructured PDFs and other enterprise documents into metadata and text to embed into retrieval systems.

A library that sits on top of RAPIDS cuDF library providing a range of benefits for processing extremely large tabular datasets, particularly those that don't fit in GPU or CPU memory. The highligh…

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing …

Python SDK for Milvus.

cuVS - a library for vector search and clustering on the GPU

Triton Backend for NVTabular

Core Utilities for NVIDIA Merlin

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation, available for both PyTorch and Tensorflow.

Merlin Models is a collection of deep learning recommender system model reference implementations

NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference i…

cuDF - GPU DataFrame Library

An implementation of a deep learning recommendation model (DLRM)

Kaggle Python docker image

Tensors and Dynamic neural networks in Python with strong GPU acceleration