rhdong's repositories
tensorflow
An Open Source Machine Learning Framework for Everyone
recommenders-addons
Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.
addons
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
all-contributors
✨ Recognize all contributors, not just the ones who push code ✨
ColossalAI
Colossal-AI: A Unified Deep Learning System for Big Model Era
cudf
cuDF - GPU DataFrame Library
cuvs
cuVS - a library for vector search and clustering on the GPU
DeepRec
DeepRec is a recommendation engine based on TensorFlow.
DPHPC23-SmokyRhino
Repository to the paper "Fast SDDMM on GPUs"
embedx
embedx 是基于 c++ 开发的、完全自研的分布式 embedding 训练和推理框架。它目前支持 图模型、深度排序、召回模型和图与排序、图与召回的联合训练模型等
faiss
A library for efficient similarity search and clustering of dense vectors.
HierarchicalKV
HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of Merlin-KV is to store key-value feature-embeddings on high-bandwidth memory (HBM) of GPUs and in host memory. It also can be used as a generic key-value storage.
HowToCook
程序员在家做饭方法指南。Programmer's guide about how to cook at home (Chinese only).
Merlin
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 in production.
models
Merlin Models is a collection of deep learning recommender system model reference implementations
NVTabular
NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale datasets used to train deep learning based recommender systems.
py-spy
Sampling profiler for Python programs
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
raft
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 high performance applications.
systems
Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature stores, nearest neighbor search, and exploration strategies) into end-to-end recommendation pipelines that can be served with Triton Inference Server.
tfra-tutorial
Tutorial on how to use dynamic embeddings from the tensorflow-recommenders-addons library
torchrec
Pytorch domain library for recommendation systems
Transformers4Rec
Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
YHs_Sample
Yinghan's Code Sample