Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 88.1k 23.6k

  2. examples Public

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

    Python 22.9k 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.7k 7k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2.1k 488

Repositories

Showing 10 of 81 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 88,063 23,639 5,000+ (85 issues need help) 1,166 Updated Mar 20, 2025
  • FBGEMM Public

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

    C++ 1,275 551 39 414 Updated Mar 20, 2025
  • executorch Public

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

    C++ 2,610 489 379 (7 issues need help) 578 Updated Mar 20, 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 90 89 222 (1 issue needs help) 185 Updated Mar 20, 2025
  • ignite Public

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

    Python 4,598 BSD-3-Clause 639 116 (34 issues need help) 45 Updated Mar 20, 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 353 123 63 40 Updated Mar 20, 2025
  • xla Public

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

    Python 2,567 506 655 (47 issues need help) 158 Updated Mar 20, 2025
  • torchtitan Public

    A PyTorch native library for large model training

    Python 3,475 BSD-3-Clause 318 79 (1 issue needs help) 54 Updated Mar 20, 2025
  • pytorch_sphinx_theme Public

    PyTorch Sphinx Theme

    CSS 99 MIT 140 21 8 Updated Mar 20, 2025
  • tensordict Public

    TensorDict is a pytorch dedicated tensor container.

    Python 898 MIT 86 36 (1 issue needs help) 29 Updated Mar 21, 2025
View all repositories