Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 84.9k 22.9k

  2. examples examples Public

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

    Python 22.5k 9.6k

  3. tutorials tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.3k 4.1k

  4. vision vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.4k 7k

  5. torchrec torchrec Public

    Pytorch domain library for recommendation systems

    Python 2k 449

Repositories

Showing 10 of 80 repositories
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    pytorch/ao’s past year of commit activity
    Python 1,656 BSD-3-Clause 184 181 (2 issues need help) 88 Updated Dec 12, 2024
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    pytorch/pytorch’s past year of commit activity
    Python 84,927 22,863 5,000+ (86 issues need help) 1,134 Updated Dec 12, 2024
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    pytorch/TensorRT’s past year of commit activity
    Python 2,618 BSD-3-Clause 351 196 31 Updated Dec 12, 2024
  • cppdocs Public

    PyTorch C++ API Documentation

    pytorch/cppdocs’s past year of commit activity
    HTML 209 34 7 3 Updated Dec 12, 2024
  • FBGEMM Public

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

    pytorch/FBGEMM’s past year of commit activity
    C++ 1,217 515 39 387 Updated Dec 12, 2024
  • torchtitan Public

    A native PyTorch Library for large model training

    pytorch/torchtitan’s past year of commit activity
    Python 2,746 BSD-3-Clause 218 45 (2 issues need help) 38 Updated Dec 12, 2024
  • executorch Public

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

    pytorch/executorch’s past year of commit activity
    C++ 2,277 388 247 (2 issues need help) 633 Updated Dec 12, 2024
  • 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.

    pytorch/test-infra’s past year of commit activity
    TypeScript 83 82 213 (1 issue needs help) 159 Updated Dec 12, 2024
  • captum Public

    Model interpretability and understanding for PyTorch

    pytorch/captum’s past year of commit activity
    Python 4,974 BSD-3-Clause 502 212 (1 issue needs help) 39 Updated Dec 12, 2024
  • 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.

    pytorch/torchx’s past year of commit activity
    Python 335 112 60 35 Updated Dec 12, 2024
View all repositories