Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 87.8k 23.6k

  2. examples Public

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

    Python 22.8k 9.6k

  3. tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.4k 4.1k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.6k 7k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2k 486

Repositories

Showing 10 of 81 repositories
  • 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 89 89 221 (1 issue needs help) 181 Updated Mar 11, 2025
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 87,782 23,568 5,000+ (83 issues need help) 1,144 Updated Mar 11, 2025
  • benchmark Public

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

    Python 914 BSD-3-Clause 302 77 69 Updated Mar 11, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    Python 1,896 BSD-3-Clause 228 218 (2 issues need help) 117 Updated Mar 11, 2025
  • executorch Public

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

    C++ 2,581 471 364 (9 issues need help) 593 Updated Mar 11, 2025
  • xla Public

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

    Python 2,544 505 644 (47 issues need help) 154 Updated Mar 11, 2025
  • torchchat Public

    Run PyTorch LLMs locally on servers, desktop and mobile

    Python 3,518 BSD-3-Clause 238 78 (6 issues need help) 14 Updated Mar 11, 2025
  • rl Public

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

    Python 2,590 MIT 341 174 (4 issues need help) 84 Updated Mar 11, 2025
  • torchrec Public

    Pytorch domain library for recommendation systems

    Python 2,047 BSD-3-Clause 485 64 297 Updated Mar 11, 2025
  • botorch Public

    Bayesian optimization in PyTorch

    Jupyter Notebook 3,193 MIT 415 60 (3 issues need help) 19 Updated Mar 11, 2025
View all repositories