pytorch
pytorch
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
torcharrow
High performance model preprocessing library on PyTorch
ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
vision
Datasets, Transforms and Models specific to Computer Vision
ELF
ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation
torchrec
Pytorch domain library for recommendation systems
TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
serve
Serve, optimize and scale PyTorch models in production
audio
Data manipulation and transformation for audio signal processing, powered by PyTorch
text
Models, data loaders and abstractions for language processing, powered by PyTorch