Give AlbumentationsX a star on GitHub — it powers this leaderboard

Star on GitHub

Packages on Leaderboard (5)

RankPackageDownloadsStarsLanguage
315torch72,489,04297,869Python
641torchvision28,235,06017,540Python
916torchaudio15,965,6242,834Python
2428torchdata2,553,2011,250
2847torchao1,891,6612,709Python

Top GitHub repositories

RepositoryDescriptionStarsLanguage
pytorch/examplesA set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.23,765Python
pytorch/tutorialsPyTorch tutorials.9,055Python
pytorch/torchtitanA PyTorch native platform for training generative AI models5,101Python
pytorch/igniteHigh-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.4,749Python
pytorch/serveServe, optimize and scale PyTorch models in production4,359Java
pytorch/executorchOn-device AI across mobile, embedded and edge for PyTorch4,333Python
pytorch/torchchatRun PyTorch LLMs locally on servers, desktop and mobile3,625Python
pytorch/textModels, data loaders and abstractions for language processing, powered by PyTorch3,565Python
pytorch/ELFELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation3,416C++
pytorch/glowCompiler for Neural Network hardware accelerators3,327C++
pytorch/rlA modular, primitive-first, python-first PyTorch library for Reinforcement Learning.3,324Python
pytorch/TensorRTPyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT2,948Python
pytorch/xlaEnabling PyTorch on XLA Devices (e.g. Google TPU)2,755C++
pytorch/android-demo-appPyTorch android examples of usage in applications1,556Java
pytorch/QNNPACKQuantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators1,547C
pytorch/FBGEMMFB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/1,534C++
pytorch/functorchfunctorch is JAX-like composable function transforms for PyTorch.1,436Jupyter Notebook
pytorch/hubSubmission to https://pytorch.org/hub/1,432Python
pytorch/glooCollective communications library with various primitives for multi-machine training.1,400C++
pytorch/extension-cppC++ extensions in PyTorch1,184Python
pytorch/cpuinfoCPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)1,155C
pytorch/torchdynamoA Python-level JIT compiler designed to make unmodified PyTorch programs faster.1,075Python
pytorch/benchmarkTorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.1,014Python
pytorch/tensordictTensorDict is a pytorch dedicated tensor container.1,010Python
pytorch/kinetoA CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters. 925HTML
pytorch/translateTranslate - a PyTorch Language Library835Python
pytorch/PiPPyPipeline Parallelism for PyTorch785Python
pytorch/helionA Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.772Python
pytorch/elasticPyTorch elastic training728Python
pytorch/torcharrowHigh performance model preprocessing library on PyTorch647Python
pytorch/ios-demo-appPyTorch iOS examples611Swift
pytorch/tvmTVM integration into PyTorch456C++
pytorch/contribImplementations of ideas from recent papers391Python
pytorch/ortAccelerate PyTorch models with ONNX Runtime368Python
pytorch/builderContinuous builder and binary build scripts for pytorch356Python
pytorch/accimagehigh performance image loading and augmenting routines mimicking PIL.Image interface319C
pytorch/tensorpipeA tensor-aware point-to-point communication primitive for machine learning284C++
pytorch/pytorch.github.ioThe website for PyTorch277HTML
pytorch/extension-ffiExamples of C extensions for PyTorch256Python
pytorch/nestedtensor[Prototype] Tools for the concurrent manipulation of variably sized Tensors.251Jupyter Notebook
pytorch/cppdocsPyTorch C++ API Documentation247HTML
pytorch/rfcsPyTorch RFCs (experimental)139
pytorch/java-demo131Jupyter Notebook
pytorch/torchdistxTorch Distributed Experimental117Python
pytorch/extension-scriptExample repository for custom C++/CUDA operators for TorchScript114Python
pytorch/test-infraThis 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.108TypeScript
pytorch/pytorch_sphinx_themePyTorch Sphinx Theme105CSS
pytorch/expecttest86Python
pytorch/maskedtensorMaskedTensors for PyTorch39Python
pytorch/pytorch-integration-testingTesting downstream libraries using pytorch release candidates15Makefile
pytorch/add-annotations-github-actionA GitHub action to run clang-tidy and annotate failures14JavaScript
pytorch/ci-infra13HCL
pytorch/ci-hudHUD for CI activity on `pytorch/pytorch`, provides a top level view for jobs to easily discern regressions13JavaScript
pytorch/probotPyTorch GitHub bot written in probot12TypeScript
pytorch/docsThis repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.11HTML
pytorch/ossci-job-dslJenkins job definitions for OSSCI10Groovy
pytorch/pytorch-ci-dockerfilesScripts for generating docker images for PyTorch CI6
pytorch/torchhub_testingRepo to test torchhub. Nothing to see here.4
pytorch/caffe2.github.ioThe caffe2 website.4HTML
pytorch/dr-ciDiagnose and remediate CI jobs3Haskell
pytorch/labeler-github-actionGitHub action for labeling issues and pull requests based on conditions3TypeScript
pytorch/alerting-infraAlerting system for the pytorch org2TypeScript
pytorch/sccachesccache is ccache with cloud storage1Rust