PyTorch

E17843

PyTorch is an open-source deep learning framework widely used for building and training neural networks, known for its dynamic computation graph and strong support for research and production in Python.

All labels observed (12)

How this entity was disambiguated

Statements (59)

Predicate Object
instanceOf Python library
deep learning framework
machine learning library
open-source software
developer Meta AI
Meta Platforms
hasComponent torch.autograd
torch.distributed
linked to: PyTorch

torch.jit
torch.nn
linked to: PyTorch

torch.optim
torch.utils.data
linked to: PyTorch

torchaudio (ecosystem)
linked to: PyTorch

torchtext (ecosystem)
torchvision (ecosystem)
hasFeature CUDA support
GPU acceleration
JIT compiler
ONNX export
TorchScript
linked to: PyTorch

autograd engine
automatic differentiation
data parallelism
distributed training
dynamic computation graph
eager execution
mixed precision training
mobile deployment support
model parallelism
multi-GPU training
quantization support
tensor computation
knownFor Pythonic interface
dynamic neural network definition
flexibility for research
license BSD 3-Clause License
linked to: BSD license
programmingLanguage C++
Python
supports inference in production
model deployment
neural network training
supportsHardware CPUs
NVIDIA GPUs
linked to: NVIDIA CUDA

TPUs (via XLA integrations)
supportsLanguage C++
Java (via extensions)
Python
R (via extensions)
targetUser data scientists
machine learning engineers
researchers
useCase computer vision
generative models
natural language processing
recommendation systems
reinforcement learning
speech recognition
time series modeling
website https://pytorch.org/

How these facts were elicited

Referenced by (50)

Full triples — surface form annotated when it differs from this entity's canonical label.

OpenAI Gym compatibleWith PyTorch
NumPy influenced PyTorch
PyTorch hasFeature TorchScript
linked to: PyTorch
PyTorch hasComponent torch.nn
linked to: PyTorch
PyTorch hasComponent torch.distributed
linked to: PyTorch
PyTorch hasComponent torch.utils.data
linked to: PyTorch
PyTorch hasComponent torchaudio (ecosystem)
linked to: PyTorch
NVIDIA DGX supports PyTorch
VGG implementedIn PyTorch
TPU supportsFramework PyTorch (via XLA or specialized backends)
linked to: PyTorch
Vertex AI supports PyTorch
torchvision framework PyTorch
subject linked to: torchvision (ecosystem)
torchvision supports TorchScript
subject linked to: torchvision (ecosystem)
linked to: PyTorch
torchvision partOf PyTorch ecosystem
subject linked to: torchvision (ecosystem)
linked to: PyTorch
torchtext compatibleWith PyTorch
subject linked to: torchtext (ecosystem)
torchtext partOf PyTorch ecosystem
subject linked to: torchtext (ecosystem)
linked to: PyTorch
torchtext maintainedBy PyTorch team
subject linked to: torchtext (ecosystem)
linked to: PyTorch
TPUs (via XLA integrations) supportsFramework PyTorch/XLA
linked to: PyTorch
Theano influenced PyTorch
RLlib supportsFramework PyTorch
Gymnasium compatibleWith PyTorch
Transformer implementedIn PyTorch
RMSProp implementedIn PyTorch
Gaudi compatibleWith PyTorch
Adam optimizer implementedIn PyTorch
WaveGlow framework PyTorch
cuDNN usedBy PyTorch
NCCL usedBy PyTorch distributed
linked to: PyTorch
Tensor Cores usedByFramework PyTorch
NVIDIA Triton Inference Server supportsFormat TorchScript
linked to: PyTorch
CycleGAN implementedIn PyTorch
AMD Instinct supportsEcosystem PyTorch (via ROCm)
subject linked to: Instinct
linked to: PyTorch
AllenNLP basedOn PyTorch
AllenNLP requires PyTorch
AllenNLP research uses PyTorch
ResNeXt implementedIn PyTorch