Wei Wang's repositories
benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
doctr
docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning.
gcp-nodes-auto-scaler
Auto scaler for Jenkins nodes setup on Google Cloud Engine instances
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch.github.io
The website for PyTorch
test-infra
This repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
torchrec
Pytorch domain library for recommendation systems
triton
Development repository for the Triton language and compiler
vision
Datasets, Transforms and Models specific to Computer Vision
xla
Enabling PyTorch on Google TPU