pytorch
OrganizationOn the leaderboard
| Rank | Repository | Stars |
|---|---|---|
| 102 | pytorch/pytorch | 98,800 |
Top repositories by stars
- pytorch/pytorch(on leaderboard)
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Python97,483 - pytorch/examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Python23,745 - pytorch/vision
Datasets, Transforms and Models specific to Computer Vision
Python17,517 - pytorch/tutorials
PyTorch tutorials.
Python9,044 - pytorch/torchtitan
A PyTorch native platform for training generative AI models
Python5,076 - pytorch/ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Python4,735 - pytorch/serve
Serve, optimize and scale PyTorch models in production
Java4,359 - pytorch/executorch
On-device AI across mobile, embedded and edge for PyTorch
Python4,277 - pytorch/torchchat
Run PyTorch LLMs locally on servers, desktop and mobile
Python3,624 - pytorch/text
Models, data loaders and abstractions for language processing, powered by PyTorch
Python3,565 - pytorch/ELF
ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation
C++3,415 - pytorch/glow
Compiler for Neural Network hardware accelerators
C++3,324 - pytorch/rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Python3,305 - pytorch/TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Python2,947 - pytorch/audio
Data manipulation and transformation for audio signal processing, powered by PyTorch
Python2,831 - pytorch/xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
C++2,749 - pytorch/ao
PyTorch native quantization and sparsity for training and inference
Python2,691 - pytorch/android-demo-app
PyTorch android examples of usage in applications
Java1,556 - pytorch/QNNPACK
Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators
C1,546 - pytorch/FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++1,535 - pytorch/functorch
functorch is JAX-like composable function transforms for PyTorch.
Jupyter Notebook1,436 - pytorch/hub
Submission to https://pytorch.org/hub/
Python1,432 - pytorch/gloo
Collective communications library with various primitives for multi-machine training.
C++1,398 - pytorch/extension-cpp
C++ extensions in PyTorch
Python1,183 - pytorch/cpuinfo
CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
C1,153 - pytorch/torchdynamo
A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
Python1,074 - pytorch/benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
Python1,014 - pytorch/tensordict
TensorDict is a pytorch dedicated tensor container.
Python1,008 - pytorch/kineto
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
HTML922 - pytorch/translate
Translate - a PyTorch Language Library
Python833 - pytorch/PiPPy
Pipeline Parallelism for PyTorch
Python785 - pytorch/helion
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
Python758 - pytorch/elastic
PyTorch elastic training
Python728 - pytorch/torcharrow
High performance model preprocessing library on PyTorch
Python646 - pytorch/ios-demo-app
PyTorch iOS examples
Swift611 - pytorch/tvm
TVM integration into PyTorch
C++456 - pytorch/contrib
Implementations of ideas from recent papers
Python392 - pytorch/ort
Accelerate PyTorch models with ONNX Runtime
Python368 - pytorch/builder
Continuous builder and binary build scripts for pytorch
Python356 - pytorch/accimage
high performance image loading and augmenting routines mimicking PIL.Image interface
C318 - pytorch/tensorpipe
A tensor-aware point-to-point communication primitive for machine learning
C++284 - pytorch/pytorch.github.io
The website for PyTorch
HTML276 - pytorch/extension-ffi
Examples of C extensions for PyTorch
Python256 - pytorch/nestedtensor
[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
Jupyter Notebook250 - pytorch/cppdocs
PyTorch C++ API Documentation
HTML246 - pytorch/rfcs
PyTorch RFCs (experimental)
138 - Jupyter Notebook131
- pytorch/torchdistx
Torch Distributed Experimental
Python117 - pytorch/extension-script
Example repository for custom C++/CUDA operators for TorchScript
Python114 - pytorch/test-infra
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.
TypeScript106 - pytorch/pytorch_sphinx_theme
PyTorch Sphinx Theme
CSS105 - Python86
- pytorch/maskedtensor
MaskedTensors for PyTorch
Python38 - pytorch/pytorch-integration-testing
Testing downstream libraries using pytorch release candidates
Makefile15 - pytorch/add-annotations-github-action
A GitHub action to run clang-tidy and annotate failures
JavaScript14 - HCL13
- pytorch/ci-hud
HUD for CI activity on `pytorch/pytorch`, provides a top level view for jobs to easily discern regressions
JavaScript13 - pytorch/probot
PyTorch GitHub bot written in probot
TypeScript12 - pytorch/docs
This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.
HTML10 - pytorch/ossci-job-dsl
Jenkins job definitions for OSSCI
Groovy10 - pytorch/pytorch-ci-dockerfiles
Scripts for generating docker images for PyTorch CI
5 - pytorch/torchhub_testing
Repo to test torchhub. Nothing to see here.
4 - pytorch/caffe2.github.io
The caffe2 website.
HTML3 - pytorch/alerting-infra
Alerting system for the pytorch org
TypeScript2 - pytorch/dr-ci
Diagnose and remediate CI jobs
Haskell2 - pytorch/labeler-github-action
GitHub action for labeling issues and pull requests based on conditions
TypeScript2 - pytorch/sccache
sccache is ccache with cloud storage
Rust1