C++ builds C++

ByteDance's Recommendation System

Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.

A machine learning compiler for GPUs, CPUs, and ML accelerators

CUDA Core Compute Libraries

Material for gpu-mode lectures

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…

cuDF - GPU DataFrame Library

compiler learning resources collect.

how to learn PyTorch and OneFlow

how to optimize some algorithm in cuda.

Fast and memory-efficient exact attention

Awesome-LLM: a curated list of Large Language Model

CUDA Templates for Linear Algebra Subroutines

PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021

Alluxio, data orchestration for analytics and machine learning in the cloud

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

LightSeq: A High Performance Library for Sequence Processing and Generation

[IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer

optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052

Transformer related optimization, including BERT, GPT

Ongoing research training transformer models at scale

Inference code for Llama models

Training and serving large-scale neural networks with auto parallelization.

ImageBind One Embedding Space to Bind Them All

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

BCC - Tools for BPF-based Linux IO analysis, networking, monitoring, and more

Awesome resources for GPUs

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.