Starred repositories
Examples for Recommenders - easy to train and deploy on accelerated infrastructure.
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…
Transcribe and translate audio to text using Whisper and DeepL.
VIM Configuration for Python / Cython / C Development
Fast and memory-efficient exact attention
HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
distributed-embeddings is a library for building large embedding based models in Tensorflow 2.
Pytorch domain library for recommendation systems
Official MegEngine implementation of CREStereo(CVPR 2022 Oral).
Source code that accompanies The CUDA Handbook.
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
An implementation of a deep learning recommendation model (DLRM)
Migrate repositories, wikis, issues and milestones from GitLab to GitHub
Multi-Agent Resource Optimization (MARO) platform is an instance of Reinforcement Learning as a Service (RaaS) for real-world resource optimization problems.
A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
Reliable Allreduce and Broadcast Interface for distributed machine learning
Papers and Codes about Deep Metric Learning/Deep Embedding
The C++ Core Guidelines are a set of tried-and-true guidelines, rules, and best practices about coding in C++
An implementation of the Raft distributed consensus protocol.
Personal code, scripts and config files for experiments