Stars
A cloud-native vector database, storage for next generation AI applications
HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of HierarchicalKV is to store key-value feature-embeddings on h…
Fast Python Collaborative Filtering for Implicit Feedback Datasets
A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.
Repository to the report "Fast SDDMM on GPUs"
Header-only C++/python library for fast approximate nearest neighbors
cuVS - a library for vector search and clustering on the GPU
A library for efficient similarity search and clustering of dense vectors.
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 …
A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.
Ongoing research training transformer models at scale
Source code for Twitter's Recommendation Algorithm
Keras Layer implementation of Attention for Sequential models
Tutorial on how to use dynamic embeddings from the tensorflow-recommenders-addons library
A memory efficient DLRM training solution using ColossalAI
[ARCHIVED] The C++ Standard Library for your entire system. See https://github.com/NVIDIA/cccl
Get started with your NVIDIA Arm HPC Developers Kit!
PyTorch and Tensorflow functional model definitions
OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature stores, nearest neighbor search, and exploration strategies) in…
The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
Making large AI models cheaper, faster and more accessible