pytorch

where the eigens are valued

Pinned Repositories

captum
Model interpretability and understanding for PyTorch
Language:Python4.9k 271 551499
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Language:Python22.4k 398 6409.5k
ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Language:Python4.5k 59 1.4k617
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:Python84.3k 1.7k 46.8k22.7k
serve
Serve, optimize and scale PyTorch models in production
Language:Java4.2k 57 1.6k863
text
Models, data loaders and abstractions for language processing, powered by PyTorch
Language:Python3.5k 309 797811
torchrec
Pytorch domain library for recommendation systems
Language:Python1.9k 33 179440
torchtune
PyTorch native finetuning library
Language:Python4.3k 47 720438
tutorials
PyTorch tutorials.
Language:Jupyter Notebook8.2k 179 8154.1k
vision
Datasets, Transforms and Models specific to Computer Vision
Language:Python16.3k 436 3.3k7k

pytorch's Repositories

pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:Python84.3k 1.7k 46.8k22.7k
pytorch/examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Language:Python22.4k 398 6409.5k
pytorch/vision
Datasets, Transforms and Models specific to Computer Vision
Language:Python16.3k 436 3.3k7k
pytorch/tutorials
PyTorch tutorials.
Language:Jupyter Notebook8.2k 179 8154.1k
pytorch/captum
Model interpretability and understanding for PyTorch
Language:Python4.9k 271 551499
pytorch/torchtune
PyTorch native finetuning library
Language:Python4.3k 47 720438
pytorch/text
Models, data loaders and abstractions for language processing, powered by PyTorch
Language:Python3.5k 309 797811
pytorch/torchchat
Run PyTorch LLMs locally on servers, desktop and mobile
Language:Python3.4k 39 331223
pytorch/botorch
Bayesian optimization in PyTorch
Language:Jupyter Notebook3.1k 52 536404
pytorch/TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Language:Python2.6k 70 1.6k351
pytorch/audio
Data manipulation and transformation for audio signal processing, powered by PyTorch
Language:Python2.5k 72 936655
pytorch/xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Language:C++2.5k 58 2.3k483
pytorch/rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Language:Python2.4k 41 634310
pytorch/executorch
On-device AI across mobile, embedded and edge for PyTorch
Language:C++2.2k 60 516365
pytorch/torchrec
Pytorch domain library for recommendation systems
Language:Python1.9k 33 179440
pytorch/tnt
A lightweight library for PyTorch training tools and utilities
Language:Python1.7k 43 71276
pytorch/ao
PyTorch native quantization and sparsity for training and inference
Language:Python1.6k 41 299177
pytorch/functorch
functorch is JAX-like composable function transforms for PyTorch.
Language:Jupyter Notebook1.4k 27 522102
pytorch/FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Language:C++1.2k 66 169502
pytorch/data
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
Language:Python1.1k 34 496152
pytorch/benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
Language:Python876 243 908287
pytorch/tensordict
TensorDict is a pytorch dedicated tensor container.
Language:Python841 30 11775
pytorch/kineto
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
Language:HTML732 25 216170
pytorch/builder
Continuous builder and binary build scripts for pytorch
Language:Shell339 25 136227
pytorch/pytorch.github.io
The website for PyTorch
Language:HTML229 24 247290
pytorch/cppdocs
PyTorch C++ API Documentation
Language:HTML209 27 1134
pytorch/multipy
torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single C++ process.
Language:C++176 15 5835
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.
Language:TypeScript83 16 3k80
pytorch/torchcodec
PyTorch video decoding
Language:Python80 6 499
pytorch/docs
This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.
Language:HTML4 5 31