pytorch 的公共 GitHub 足迹

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Datasets, Transforms and Models specific to Computer Vision

PyTorch tutorials.

A PyTorch native platform for training generative AI models

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

On-device AI across mobile, embedded and edge for PyTorch

Serve, optimize and scale PyTorch models in production

Run PyTorch LLMs locally on servers, desktop and mobile

Models, data loaders and abstractions for language processing, powered by PyTorch

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation

Compiler for Neural Network hardware accelerators

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Data manipulation and transformation for audio signal processing, powered by PyTorch

PyTorch native quantization and sparsity for training and inference

Enabling PyTorch on XLA Devices (e.g. Google TPU)

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

PyTorch android examples of usage in applications

Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators

functorch is JAX-like composable function transforms for PyTorch.

Submission to https://pytorch.org/hub/

Collective communications library with various primitives for multi-machine training.

C++ extensions in PyTorch

CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

TensorDict is a pytorch dedicated tensor container.

A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

Translate - a PyTorch Language Library

Pipeline Parallelism for PyTorch

PyTorch elastic training

High performance model preprocessing library on PyTorch

PyTorch iOS examples

TVM integration into PyTorch

Implementations of ideas from recent papers

Accelerate PyTorch models with ONNX Runtime

Continuous builder and binary build scripts for pytorch

high performance image loading and augmenting routines mimicking PIL.Image interface

A tensor-aware point-to-point communication primitive for machine learning

The website for PyTorch

Examples of C extensions for PyTorch

PyTorch C++ API Documentation

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

PyTorch RFCs (experimental)

此仓库未提供描述。

Torch Distributed Experimental

Example repository for custom C++/CUDA operators for TorchScript

This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

PyTorch Sphinx Theme

此仓库未提供描述。

MaskedTensors for PyTorch

Testing downstream libraries using pytorch release candidates

此仓库未提供描述。

Developer blog for PyTorch

A GitHub action to run clang-tidy and annotate failures

HUD for CI activity on `pytorch/pytorch`, provides a top level view for jobs to easily discern regressions

PyTorch GitHub bot written in probot

This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.

Jenkins job definitions for OSSCI

Scripts for generating docker images for PyTorch CI

Curated distribution of PyTorch-compatible Python sub-libraries (the PyTorch annex)

The caffe2 website.

Repo to test torchhub. Nothing to see here.

Test for Cross-Repository CI Relay for PyTorch Out-of-Tree Backend

GitHub action for labeling issues and pull requests based on conditions

Diagnose and remediate CI jobs

Alerting system for the pytorch org

sccache is ccache with cloud storage