pytorch/ao

PyTorch native quantization and sparsity for training and inference

Language: Python - Size: 41.5 MB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 2,309 - Forks: 328

pytorch/test-infra

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.

Language: TypeScript - Size: 1.24 GB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 97 - Forks: 102

pytorch/TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Language: Python - Size: 175 MB - Last synced at: 2 days ago - Pushed at: 3 days ago - Stars: 2,848 - Forks: 372

pytorch/helion

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

Language: Python - Size: 2.57 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 284 - Forks: 25

pytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Language: Python - Size: 1.03 GB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 92,871 - Forks: 25,154

pytorch/executorch

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

Language: Python - Size: 241 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 3,184 - Forks: 654

pytorch/vision

Datasets, Transforms and Models specific to Computer Vision

Language: Python - Size: 1.17 GB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 17,111 - Forks: 7,145

pytorch/torchrec

Pytorch domain library for recommendation systems

Language: Python - Size: 22.7 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 2,328 - Forks: 558

pytorch/examples

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

Language: Python - Size: 40.8 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 23,352 - Forks: 9,721

pytorch/tnt

A lightweight library for PyTorch training tools and utilities

Language: Python - Size: 8.59 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 1,707 - Forks: 293

pytorch/ignite

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

Language: Python - Size: 52.8 MB - Last synced at: 6 days ago - Pushed at: 2 months ago - Stars: 4,691 - Forks: 652

pytorch/captum

Model interpretability and understanding for PyTorch

Language: Python - Size: 310 MB - Last synced at: 6 days ago - Pushed at: 23 days ago - Stars: 5,388 - Forks: 541

pytorch/tensorpipe

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

Language: C++ - Size: 2.23 MB - Last synced at: 6 days ago - Pushed at: 23 days ago - Stars: 265 - Forks: 76

pytorch/hub

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

Language: Python - Size: 25.2 MB - Last synced at: 6 days ago - Pushed at: over 1 year ago - Stars: 1,426 - Forks: 247

pytorch/tutorials

PyTorch tutorials.

Language: Jupyter Notebook - Size: 7.74 GB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 8,761 - Forks: 4,228

pytorch/torchtitan

A PyTorch native platform for training generative AI models

Language: Python - Size: 93.7 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 4,324 - Forks: 495

pytorch/torchcodec

PyTorch media decoding and encoding

Language: Python - Size: 56 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 686 - Forks: 56

pytorch/PiPPy 📦

Pipeline Parallelism for PyTorch

Language: Python - Size: 4.01 MB - Last synced at: 6 days ago - Pushed at: about 1 year ago - Stars: 777 - Forks: 87

pytorch/tensordict

TensorDict is a pytorch dedicated tensor container.

Language: Python - Size: 143 MB - Last synced at: 6 days ago - Pushed at: 19 days ago - Stars: 961 - Forks: 100

pytorch/extension-cpp

C++ extensions in PyTorch

Language: Python - Size: 46.9 KB - Last synced at: 6 days ago - Pushed at: 2 months ago - Stars: 1,141 - Forks: 242

pytorch/rl

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

Language: Python - Size: 209 MB - Last synced at: 6 days ago - Pushed at: 12 days ago - Stars: 3,025 - Forks: 402

pytorch/text

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

Language: Python - Size: 47.3 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 3,558 - Forks: 815

pytorch/ELF 📦

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

Language: C++ - Size: 6.13 MB - Last synced at: 6 days ago - Pushed at: about 6 years ago - Stars: 3,399 - Forks: 569

pytorch/torchft

Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)

Language: Python - Size: 2.74 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 392 - Forks: 43

pytorch/FBGEMM

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

Language: C++ - Size: 32.1 MB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 1,429 - Forks: 638

pytorch/ios-demo-app 📦

PyTorch iOS examples

Language: Swift - Size: 129 MB - Last synced at: 6 days ago - Pushed at: about 1 year ago - Stars: 610 - Forks: 193

pytorch/botorch

Bayesian optimization in PyTorch

Language: Jupyter Notebook - Size: 127 MB - Last synced at: 6 days ago - Pushed at: 10 days ago - Stars: 3,336 - Forks: 436

pytorch/torchtune

PyTorch native post-training library

Language: Python - Size: 62.3 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 5,446 - Forks: 668

pytorch/cpuinfo

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

Language: C - Size: 4.95 MB - Last synced at: 6 days ago - Pushed at: 30 days ago - Stars: 1,115 - Forks: 371

pytorch/torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

Language: Python - Size: 9.02 MB - Last synced at: 6 days ago - Pushed at: 26 days ago - Stars: 3,607 - Forks: 251

pytorch/pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

Language: Shell - Size: 406 KB - Last synced at: 6 days ago - Pushed at: 8 days ago - Stars: 11 - Forks: 18

pytorch/kineto

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

Language: HTML - Size: 43.8 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 856 - Forks: 200

pytorch/xla

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

Language: Python - Size: 90.6 MB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 2,661 - Forks: 563

pytorch/torchx

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

Language: Python - Size: 34.8 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 387 - Forks: 143

pytorch/android-demo-app 📦

PyTorch android examples of usage in applications

Language: Java - Size: 130 MB - Last synced at: 6 days ago - Pushed at: about 1 year ago - Stars: 1,543 - Forks: 627

pytorch/opacus

Training PyTorch models with differential privacy

Language: Jupyter Notebook - Size: 7.49 MB - Last synced at: 6 days ago - Pushed at: 17 days ago - Stars: 1,844 - Forks: 374

pytorch/audio

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

Language: Python - Size: 1.62 GB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 2,723 - Forks: 719

pytorch/ci-infra

Language: HCL - Size: 5.06 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 14 - Forks: 1

pytorch/multipy

torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single C++ process.

Language: C++ - Size: 2.96 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 180 - Forks: 36

pytorch/benchmark

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

Language: Python - Size: 229 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 982 - Forks: 318

pytorch/torcharrow 📦

High performance model preprocessing library on PyTorch

Language: Python - Size: 11.3 MB - Last synced at: 6 days ago - Pushed at: over 1 year ago - Stars: 646 - Forks: 81

pytorch/QNNPACK 📦

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

Language: C - Size: 522 KB - Last synced at: 6 days ago - Pushed at: about 6 years ago - Stars: 1,542 - Forks: 221

pytorch/pytorch.github.io

The website for PyTorch

Language: HTML - Size: 6.92 GB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 260 - Forks: 312

pytorch/tlparse

TORCH_LOGS parser for PT2

Language: Rust - Size: 5.96 MB - Last synced at: 6 days ago - Pushed at: 10 days ago - Stars: 57 - Forks: 17

pytorch/functorch 📦

functorch is JAX-like composable function transforms for PyTorch.

Language: Jupyter Notebook - Size: 22.1 MB - Last synced at: 6 days ago - Pushed at: 18 days ago - Stars: 1,437 - Forks: 105

pytorch/rfcs

PyTorch RFCs (experimental)

Size: 738 KB - Last synced at: 6 days ago - Pushed at: 3 months ago - Stars: 135 - Forks: 83

pytorch/torchdynamo 📦

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

Language: Python - Size: 6.2 MB - Last synced at: 6 days ago - Pushed at: over 1 year ago - Stars: 1,060 - Forks: 128

pytorch/gloo

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

Language: C++ - Size: 1.54 MB - Last synced at: 6 days ago - Pushed at: 17 days ago - Stars: 1,346 - Forks: 340

pytorch/cppdocs

PyTorch C++ API Documentation

Language: HTML - Size: 2.41 GB - Last synced at: 6 days ago - Pushed at: 11 days ago - Stars: 234 - Forks: 37

pytorch/builder 📦

Continuous builder and binary build scripts for pytorch

Language: Python - Size: 3.89 MB - Last synced at: 6 days ago - Pushed at: 26 days ago - Stars: 354 - Forks: 229

pytorch/data

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

Language: Python - Size: 17.7 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 1,221 - Forks: 168

pytorch/workshops

This is a repository for all workshop related materials.

Language: Jupyter Notebook - Size: 179 MB - Last synced at: 6 days ago - Pushed at: over 1 year ago - Stars: 228 - Forks: 87

pytorch/serve 📦

Serve, optimize and scale PyTorch models in production

Language: Java - Size: 120 MB - Last synced at: 6 days ago - Pushed at: about 1 month ago - Stars: 4,347 - Forks: 887

pytorch/glow 📦

Compiler for Neural Network hardware accelerators

Language: C++ - Size: 74.2 MB - Last synced at: 6 days ago - Pushed at: over 1 year ago - Stars: 3,310 - Forks: 698

pytorch/torcheval

A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.

Language: Python - Size: 3.63 MB - Last synced at: 6 days ago - Pushed at: 15 days ago - Stars: 236 - Forks: 55

pytorch/torchsnapshot

A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.

Language: Python - Size: 5.82 MB - Last synced at: 6 days ago - Pushed at: 3 months ago - Stars: 159 - Forks: 44

pytorch/accimage 📦

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

Language: C - Size: 702 KB - Last synced at: 6 days ago - Pushed at: about 4 years ago - Stars: 319 - Forks: 38

pytorch/translate 📦

Translate - a PyTorch Language Library

Language: Python - Size: 2.43 MB - Last synced at: 6 days ago - Pushed at: over 2 years ago - Stars: 835 - Forks: 194

pytorch/ort

Accelerate PyTorch models with ONNX Runtime

Language: Python - Size: 12.7 MB - Last synced at: 6 days ago - Pushed at: 6 months ago - Stars: 363 - Forks: 53

pytorch/expecttest

Language: Python - Size: 65.4 KB - Last synced at: 6 days ago - Pushed at: 3 months ago - Stars: 83 - Forks: 9

pytorch/maskedtensor 📦

MaskedTensors for PyTorch

Language: Python - Size: 3.2 MB - Last synced at: 6 days ago - Pushed at: about 3 years ago - Stars: 38 - Forks: 12

pytorch/java-demo 📦

Language: Jupyter Notebook - Size: 116 KB - Last synced at: 6 days ago - Pushed at: about 4 years ago - Stars: 129 - Forks: 23

pytorch/csprng

Cryptographically secure pseudorandom number generators for PyTorch

Language: Batchfile - Size: 431 KB - Last synced at: 6 days ago - Pushed at: 3 months ago - Stars: 121 - Forks: 31

pytorch/elastic 📦

PyTorch elastic training

Language: Python - Size: 3.84 MB - Last synced at: 6 days ago - Pushed at: about 3 years ago - Stars: 729 - Forks: 100

pytorch/pytorch_sphinx_theme

PyTorch Sphinx Theme

Language: CSS - Size: 19.9 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 101 - Forks: 143

pytorch/torchdistx 📦

Torch Distributed Experimental

Language: Python - Size: 1.42 MB - Last synced at: 6 days ago - Pushed at: about 1 year ago - Stars: 117 - Forks: 33

pytorch/hydra-torch

Configuration classes enabling type-safe PyTorch configuration for Hydra apps

Language: Python - Size: 10.9 MB - Last synced at: 6 days ago - Pushed at: almost 3 years ago - Stars: 221 - Forks: 17

pytorch/contrib 📦

Implementations of ideas from recent papers

Language: Python - Size: 94.7 KB - Last synced at: 6 days ago - Pushed at: over 4 years ago - Stars: 392 - Forks: 44

pytorch/tvm 📦

TVM integration into PyTorch

Language: C++ - Size: 586 KB - Last synced at: 6 days ago - Pushed at: over 5 years ago - Stars: 453 - Forks: 64

pytorch/extension-ffi 📦

Examples of C extensions for PyTorch

Language: Python - Size: 9.77 KB - Last synced at: 6 days ago - Pushed at: over 2 years ago - Stars: 256 - Forks: 70

pytorch/nestedtensor 📦

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

Language: Jupyter Notebook - Size: 12.9 MB - Last synced at: 6 days ago - Pushed at: almost 3 years ago - Stars: 251 - Forks: 27

pytorch/docs

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

Language: HTML - Size: 2.4 GB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 9 - Forks: 4

pytorch/extension-script 📦

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

Language: Python - Size: 13.7 KB - Last synced at: 6 days ago - Pushed at: about 3 years ago - Stars: 114 - Forks: 23

pytorch/caffe2.github.io Fork of caffe2/caffe2.github.io 📦

The caffe2 website.

Language: HTML - Size: 367 MB - Last synced at: 6 days ago - Pushed at: over 6 years ago - Stars: 3 - Forks: 8

pytorch/pytorch-ci-dockerfiles 📦

Scripts for generating docker images for PyTorch CI

Size: 150 KB - Last synced at: 6 days ago - Pushed at: over 5 years ago - Stars: 4 - Forks: 8

pytorch/labeler-github-action 📦

GitHub action for labeling issues and pull requests based on conditions

Language: TypeScript - Size: 531 KB - Last synced at: 6 days ago - Pushed at: almost 5 years ago - Stars: 2 - Forks: 7

pytorch/probot 📦

PyTorch GitHub bot written in probot

Language: TypeScript - Size: 16.4 MB - Last synced at: 6 days ago - Pushed at: over 3 years ago - Stars: 12 - Forks: 13

pytorch/ci-hud 📦

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

Language: JavaScript - Size: 34.9 MB - Last synced at: 6 days ago - Pushed at: over 3 years ago - Stars: 13 - Forks: 24

pytorch/add-annotations-github-action 📦

A GitHub action to run clang-tidy and annotate failures

Language: JavaScript - Size: 16.4 MB - Last synced at: 16 days ago - Pushed at: about 2 years ago - Stars: 14 - Forks: 12

pytorch/ossci-job-dsl 📦

Jenkins job definitions for OSSCI

Language: Groovy - Size: 1020 KB - Last synced at: 6 days ago - Pushed at: over 3 years ago - Stars: 10 - Forks: 33

pytorch/torchhub_testing 📦

Repo to test torchhub. Nothing to see here.

Size: 5.86 KB - Last synced at: 6 days ago - Pushed at: over 2 years ago - Stars: 4 - Forks: 1

pytorch/sccache Fork of mozilla/sccache 📦

sccache is ccache with cloud storage

Language: Rust - Size: 2.73 MB - Last synced at: 6 days ago - Pushed at: about 5 years ago - Stars: 1 - Forks: 2

pytorch/dr-ci 📦

Diagnose and remediate CI jobs

Language: Haskell - Size: 2.05 MB - Last synced at: 6 days ago - Pushed at: over 4 years ago - Stars: 2 - Forks: 8