Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 89.4k 24k

  2. examples Public

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

    Python 23k 9.6k

  3. tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.5k 4.1k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.8k 7k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2.1k 492

Repositories

Showing 10 of 81 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 89,355 23,957 5,000+ (84 issues need help) 1,310 Updated Apr 24, 2025
  • torchx Public

    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.

    Python 360 129 66 45 Updated Apr 24, 2025
  • executorch Public

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

    Python 2,754 528 425 (14 issues need help) 579 Updated Apr 24, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    Python 1,991 BSD-3-Clause 248 230 (2 issues need help) 113 Updated Apr 24, 2025
  • tensordict Public

    TensorDict is a pytorch dedicated tensor container.

    Python 912 MIT 88 35 (1 issue needs help) 30 Updated Apr 24, 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 91 92 218 (1 issue needs help) 195 Updated Apr 24, 2025
  • audio Public

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

    Python 2,648 BSD-2-Clause 687 226 (12 issues need help) 58 Updated Apr 24, 2025
  • rl Public

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

    Python 2,687 MIT 361 180 (4 issues need help) 93 Updated Apr 24, 2025
  • FBGEMM Public

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

    C++ 1,304 566 41 430 Updated Apr 24, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    Python 2,731 BSD-3-Clause 362 208 25 Updated Apr 24, 2025
View all repositories