Bowen Bao's repositories
benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
ControlNet
Let us control diffusion models!
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
HRNet-Image-Classification
Train the HRNet model on ImageNet
jschema-to-python
Generate source code for a set of Python classes from a JSON schema.
llama
Inference code for LLaMA models
mlagility
Machine Learning Agility (MLAgility) benchmark and benchmarking tools
onnx-docker
Dockerfiles and scripts for ONNX container images
onnx-mlir
Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
onnx-script
ONNX Script enables developers to naturally author ONNX functions and models using a subset of Python. ⚠️ ONNX Script is in very early and active development and is not ready for production.
onnxmltools
ONNXMLTools enables conversion of models to ONNX
onnxruntime
ONNX Runtime
optimum
🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools
pytorch-pretrained-BERT
A PyTorch implementation of Google AI's BERT model provided with Google's pre-trained models, examples and utilities.
pytorch-ssd
MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
sarif-vscode-extension
SARIF Microsoft Visual Studio Code extension
sklearn-onnx
Convert scikit-learn models and pipelines to ONNX
tacotron2
Tacotron 2 - PyTorch implementation with faster-than-realtime inference
triton
Development repository for the Triton language and compiler
UGATIT-pytorch
Official PyTorch implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation