jerryzh168

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

ao

torchao: PyTorch Architecture Optimization (AO). A repository to host AO techniques and performant kernels that work with PyTorch.

executorch-1

End-to-end solution for enabling on-device AI across mobile and edge devices for PyTorch models

FasterTransformer

Transformer related optimization, including BERT, GPT

synchrobench

A benchmark to compare synchronization techniques for multicore programming

15688-proj

Appendix Notebook for 15688 course project

allennlp

An open-source NLP research library, built on PyTorch.

Background-Matting

Background Matting: The World is Your Green Screen

benchmark

caffe2

Caffe2 is a cross-platform framework made with expression, speed, and modularity in mind.

caffe2-benchmarking

Benchmark Caffe2 performance and accuracy

ClassyVision-1

An end-to-end PyTorch framework for image and video classification

coco

MS COCO API - http://mscoco.org/

ComputeLibrary

The ARM Computer Vision and Machine Learning library is a set of functions optimised for both ARM CPUs and GPUs using SIMD technologies.

d2go

D2Go is a toolkit for efficient deep learning

detectron2-1

Detectron2 is FAIR's next-generation platform for object detection, segmentation and other visual recognition tasks.

emacs.d

An Emacs configuration bundle with batteries included

hackerrank

My Hackerrank Solutions

mobile-vision

Mobile vision models and code

pytorch.github.io

The website for PyTorch

rfcs

PyTorch RFCs (experimental)

torchrec

Pytorch domain library for recommendation systems

torchtune

A Native-PyTorch Library for LLM Fine-tuning

translate

Translate - a PyTorch Language Library

tutorials

PyTorch tutorials.

vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.