Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 85.4k 23k

  2. examples Public

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

    Python 22.6k 9.6k

  3. tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.3k 4.1k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.4k 7k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2k 455

Repositories

Showing 10 of 80 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

  • executorch Public

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

    C++ 2,351 409 271 (2 issues need help) 661 Updated Jan 2, 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 83 84 219 (1 issue needs help) 168 Updated Jan 1, 2025
  • docs Public

    This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.

    HTML 7 BSD-3-Clause 1 2 0 Updated Jan 1, 2025
  • functorch Public

    functorch is JAX-like composable function transforms for PyTorch.

    Jupyter Notebook 1,402 BSD-3-Clause 102 232 (1 issue needs help) 18 Updated Jan 1, 2025
  • serve Public

    Serve, optimize and scale PyTorch models in production

    Java 4,253 Apache-2.0 866 378 (20 issues need help) 47 Updated Jan 1, 2025
  • torchcodec Public

    PyTorch video decoding

    Python 184 BSD-3-Clause 14 38 9 Updated Jan 1, 2025
  • torchtune Public

    PyTorch native post-training library

    Python 4,530 BSD-3-Clause 471 240 (6 issues need help) 52 Updated Jan 1, 2025
  • FBGEMM Public

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

    C++ 1,231 520 40 380 Updated Jan 1, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    Python 2,630 BSD-3-Clause 355 190 15 Updated Jan 1, 2025