Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 90.8k 24.4k

  2. examples Public

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

    Python 23.1k 9.7k

  3. tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.6k 4.2k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.9k 7.1k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2.2k 525

Repositories

Showing 10 of 82 repositories
  • executorch Public

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

    Python 2,951 592 554 (45 issues need help) 512 Updated Jun 13, 2025
  • torchtitan Public

    A PyTorch native platform for training generative AI models

    Python 3,916 BSD-3-Clause 392 104 (1 issue needs help) 73 Updated Jun 13, 2025
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 90,784 24,431 5,000+ (90 issues need help) 1,372 Updated Jun 13, 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 94 100 229 (1 issue needs help) 210 Updated Jun 13, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    Python 2,781 BSD-3-Clause 363 213 23 Updated Jun 13, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    Python 2,104 BSD-3-Clause 283 255 (2 issues need help) 140 Updated Jun 13, 2025
  • captum Public

    Model interpretability and understanding for PyTorch

    Python 5,262 BSD-3-Clause 523 218 (1 issue needs help) 41 Updated Jun 13, 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 366 132 71 51 Updated Jun 13, 2025
  • torchrec Public

    Pytorch domain library for recommendation systems

    Python 2,237 BSD-3-Clause 524 85 337 Updated Jun 13, 2025
  • xla Public

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

    Python 2,622 536 648 (47 issues need help) 188 Updated Jun 13, 2025
View all repositories

Top languages

Loading…