Puzzles for learning Triton

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

FlashInfer: Kernel Library for LLM Serving

Fork of LLVM to support AMD AIEngine processors

An open-source image signal processing (ISP) pipeline implemented by C++

Hypersim: A Photorealistic Synthetic Dataset for Holistic Indoor Scene Understanding

depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.

🏞️ PicX 是一款基于 GitHub API 开发的图床工具，提供图片上传托管、生成图片链接和常用图片工具箱服务。

Quantized Attention that achieves speedups of 2.1x and 2.7x compared to FlashAttention2 and xformers, respectively, without lossing end-to-end metrics across various models.

Helpful tools and examples for working with flex-attention

Depth Pro: Sharp Monocular Metric Depth in Less Than a Second.

Deep insight tensorrt, including but not limited to qat, ptq, plugin, triton_inference, cuda

The Free and Open Source Cross Platform YUV Viewer with an advanced analytics toolset

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation

A wheel-mounted MEMS IMU-based dead reckoning system.

CameraSDK-Cpp is a C++ library to control Insta360 cameras.

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Compass Apache TVM is enhanced based on the Apache TVM for wide range of Neural Network (NN) models quick support, optimization and heterogeneous execution.

The official implementation of "NAS-BNN: Neural Architecture Search for Binary Neural Networks"

Everything in Torch Fx

A easy tool for generating Tensor Program from Torch(besd on Torch FX & TVM Relax)

TVM Relay IR Visualization Tool (TVM 可视化工具)

PyTorch native quantization and sparsity for training and inference

A pytorch quantization backend for optimum

Training library for local feature detection and matching