pytorch

where the eigens are valued

Pinned Repositories

captum
Model interpretability and understanding for PyTorch
Language:Python4.6k 244 518464
ELF
ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation
Language:C++3.4k 189 132567
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Language:Python21.8k 398 6299.5k
ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Language:Python4.5k 60 1.4k606
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:Python78.6k 1.7k 42k21.2k
serve
Serve, optimize and scale PyTorch models in production
Language:Java4k 58 1.6k808
text
Models, data loaders and abstractions for language processing, powered by PyTorch
Language:Python3.4k 288 790812
torchrec
Pytorch domain library for recommendation systems
Language:Python1.7k 27 139362
tutorials
PyTorch tutorials.
Language:Jupyter Notebook7.8k 178 7363.9k
vision
Datasets, Transforms and Models specific to Computer Vision
Language:Python15.5k 413 3.1k6.8k

pytorch's Repositories

pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:Python78.6k 1.7k 42k21.2k
pytorch/vision
Datasets, Transforms and Models specific to Computer Vision
Language:Python15.5k 413 3.1k6.8k
pytorch/tutorials
PyTorch tutorials.
Language:Jupyter Notebook7.8k 178 7363.9k
pytorch/serve
Serve, optimize and scale PyTorch models in production
Language:Java4k 58 1.6k808
pytorch/text
Models, data loaders and abstractions for language processing, powered by PyTorch
Language:Python3.4k 288 790812
pytorch/torchtune
A Native-PyTorch Library for LLM Fine-tuning
Language:Python3.2k 38 283221
pytorch/botorch
Bayesian optimization in PyTorch
Language:Jupyter Notebook3k 53 477374
pytorch/audio
Data manipulation and transformation for audio signal processing, powered by PyTorch
Language:Python2.4k 73 910633
pytorch/TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Language:Python2.4k 70 1.4k329
pytorch/xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Language:C++2.3k 57 1.8k404
pytorch/rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Language:Python1.9k 40 525256
pytorch/torchrec
Pytorch domain library for recommendation systems
Language:Python1.7k 27 139362
pytorch/tnt
A lightweight library for PyTorch training tools and utilities
Language:Python1.6k 46 68262
pytorch/opacus
Training PyTorch models with differential privacy
Language:Jupyter Notebook1.6k 43 284318
pytorch/functorch
functorch is JAX-like composable function transforms for PyTorch.
Language:Jupyter Notebook1.4k 27 520102
pytorch/executorch
On-device AI across mobile, embedded and edge for PyTorch
Language:C++1.3k 53 238206
pytorch/torchtitan
A native PyTorch Library for large model training
Language:Python1.1k 27 80104
pytorch/FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Language:C++1.1k 52 137421
pytorch/data
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
Language:Python1.1k 35 475140
pytorch/benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
Language:Python790 224 842247
pytorch/PiPPy
Pipeline Parallelism for PyTorch
Language:Python634 36 24575
pytorch/tensordict
TensorDict is a pytorch dedicated tensor container.
Language:Python605 27 8554
pytorch/ort
Accelerate PyTorch models with ONNX Runtime
Language:Python346 24 3753
pytorch/builder
Continuous builder and binary build scripts for pytorch
Language:Shell320 25 118208
pytorch/torchx
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.
Language:Python303 19 17996
pytorch/ao
Native PyTorch library for quantization and sparsity
Language:Python241 16 3133
pytorch/pytorch.github.io
The website for PyTorch
Language:HTML211 24 237276
pytorch/cppdocs
PyTorch C++ API Documentation
Language:HTML194 26 1134
pytorch/test-infra
This repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
Language:TypeScript68 14 2.9k63
pytorch/docs
This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.
Language:HTML3