NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
Role in this project:
ML Engineer Contributions:1 release, 40 reviews, 31 commits in 2 years 4 months
Contributions summary:Xiaodong primarily contributed to optimizing and improving the performance of deep learning inference within the NVIDIA TensorRT framework. Their work included enhancing the engine build process for BERT models, specifically by disabling certain tactics and integrating timing caches. They also addressed bugs and made code adjustments related to the use of full masks for specific hardware and data types, ensuring optimal performance across different architectures. In addition, they provided updates and fixes related to pytorch-quantization.
deep-learninginferencenvidiatensorrtgpu-acceleration
FlashInfer: Kernel Library for LLM Serving
Contributions:58 pushes, 20 branches in 2 months