Location:where the eigens are valued
Github PK Tool:Github PK Tool
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Datasets, Transforms and Models specific to Computer Vision
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Model interpretability and understanding for PyTorch
Data loaders and abstractions for text and NLP
Compiler for Neural Network hardware accelerators
Serve, optimize and scale PyTorch models in production
Bayesian optimization in PyTorch
Data manipulation and transformation for audio signal processing, powered by PyTorch
Enabling PyTorch on Google TPU
PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT
Training PyTorch models with differential privacy
Pytorch domain library for recommendation systems
functorch is JAX-like composable function transforms for PyTorch.
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
PyTorch Live is an easy to use library of tools for creating on-device ML demos on Android and iOS. With Live, you can build a working mobile ML prototype in minutes.
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
The website for PyTorch
A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
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.
PyTorch C++ API Documentation
Cryptographically secure pseudorandom number generators for PyTorch
Pipeline Parallelism for PyTorch
MaskedTensors for PyTorch
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.