numpee

git-flight-rules

Flight rules for git

Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything

flash-attention

Fast and memory-efficient exact attention

triton

Development repository for the Triton language and compiler

TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.

xformers

Hackable and optimized Transformers building blocks, supporting a composable construction.

fiftyone

The open-source tool for building high-quality datasets and computer vision models

GPU-Puzzles

Solve puzzles. Learn CUDA.

Mask2Former

Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"

detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

ml-fastvit

This repository contains the official implementation of the research paper, "FastViT: A Fast Hybrid Vision Transformer using Structural Reparameterization" ICCV 2023

kernl

Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.

awesome-segment-anything

Tracking and collecting papers/projects/others related to Segment Anything.

MaskDINO

[CVPR 2023] Official implementation of the paper "Mask DINO: Towards A Unified Transformer-based Framework for Object Detection and Segmentation"

ToMe

A method to increase the speed and lower the memory footprint of existing vision transformers.

Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training

FasterViT

[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention

hiera

Hiera: A fast, powerful, and simple hierarchical vision transformer.

Awesome-GPU

Awesome resources for GPUs

BEVFormer_tensorrt

BEVFormer inference on TensorRT, including INT8 Quantization and Custom TensorRT Plugins (float/half/half2/int8).

CUDA_gemm

A simple high performance CUDA GEMM implementation.

trident

A performance library for machine learning applications.

ClothWild_RELEASE

[ECCV 2022] This repo is official PyTorch implementation of 3D Clothed Human Reconstruction in the Wild.

Partial_Distance_Correlation

This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022

mask-auto-labeler

Focus-DETR-mindspore

[ICCV 2023] Official implementation of the paper "Less is More: Focus Attention for Efficient DETR"

uncertainty_retrieval

ICLR‘24 Offical Implementation of Composed Image Retrieval with Text Feedback via Multi-grained Uncertainty Regularization

CKA.pytorch

A PyTorch implementation of Centered Kernel Alignment (CKA) with GPU acceleration.

FedMLB

Official implementation of "Multi-Level Branched Regularization for Federated Learning" in ICML 2022

TCD

Code for "Class-Incremental Learning for Action Recognition in Videos", ICCV 2021