Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 93.8k 25.5k

  2. examples Public

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

    Python 23.4k 9.8k

  3. tutorials Public

    PyTorch tutorials.

    Python 8.8k 4.3k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 17.2k 7.2k

Repositories

Showing 10 of 67 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 93,766 25,502 5,000+ (81 issues need help) 1,476 Updated Oct 9, 2025
  • helion Public

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

    Python 333 BSD-3-Clause 36 55 31 Updated Oct 9, 2025
  • docs Public

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

    HTML 9 BSD-3-Clause 5 2 0 Updated Oct 9, 2025
  • test-infra Public

    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.

    TypeScript 102 105 217 (1 issue needs help) 241 Updated Oct 9, 2025
  • FBGEMM Public

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

    C++ 1,448 657 54 545 Updated Oct 9, 2025
  • executorch Public

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

    Python 3,289 683 763 (38 issues need help) 676 Updated Oct 9, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    Python 2,394 344 321 (2 issues need help) 179 Updated Oct 9, 2025
  • torchtitan Public

    A PyTorch native platform for training generative AI models

    Python 4,510 BSD-3-Clause 555 120 (1 issue needs help) 141 Updated Oct 8, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    Python 2,868 BSD-3-Clause 370 246 31 Updated Oct 8, 2025
  • benchmark Public

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

    Python 986 BSD-3-Clause 322 80 78 Updated Oct 8, 2025
View all repositories