PyTorch (GPU)

GPTKB entity

Statements (98)
Predicate Object
gptkbp:instanceOf gptkb:convolutional_neural_network
gptkbp:competitor gptkb:TensorFlow
gptkb:MXNet
gptkb:JAX
gptkbp:component autograd
torch.Tensor
torch.nn
torch.optim
torch.utils.data
gptkbp:developedBy gptkb:Meta_AI
gptkbp:documentation https://pytorch.org/docs/stable/index.html
gptkbp:feature gptkb:Dynamo
gptkb:PyTorch_Lightning
gptkb:TorchAudio
gptkb:TorchServe
gptkb:TorchText
gptkb:TorchVision
gptkb:TorchScript
gptkb:TorchDynamo
modular design
model deployment
mobile support
distributed training
Inductor
integration with NumPy
automatic differentiation
mixed precision training
model export
model pruning
model quantization
extensive ecosystem
Jupyter Notebook support
ONNX export
TorchMetrics
TorchData
TorchRL
TorchRec
TorchX
TorchInductor
torch.jit
C++ frontend
FX (Functional Transformations)
Hugging Face Transformers integration
TorchCompile
custom CUDA extensions
easy debugging
torch.cuda
torch.distributed
torch.fx
torch.hub
torch.multiprocessing
torch.onnx
torch.package
torch.profiler
torch.quantization
torch.sparse
torch.utils.bottleneck
torch.utils.checkpoint
torch.utils.tensorboard
gptkbp:firstReleased 2016
https://www.w3.org/2000/01/rdf-schema#label PyTorch (GPU)
gptkbp:latestReleaseVersion 2.3.0
gptkbp:license gptkb:BSD_license
gptkbp:notableUser gptkb:Amazon
gptkb:DeepMind
gptkb:Google
gptkb:MIT
gptkb:Microsoft
gptkb:OpenAI
gptkb:Salesforce
gptkb:Stanford_University
gptkb:Tesla
gptkb:Uber
gptkb:NVIDIA
gptkb:Meta
gptkbp:openSource true
gptkbp:platform gptkb:Windows
gptkb:macOS
gptkb:Linux
gptkbp:programmingLanguage gptkb:Python
gptkb:C++
gptkbp:repository https://github.com/pytorch/pytorch
gptkbp:supports gptkb:CUDA
gptkb:NVIDIA_GPUs
gptkb:ROCm
gptkb:AMD_GPUs
GPU acceleration
dynamic computation graphs
static computation graphs (TorchScript)
gptkbp:usedBy gptkb:industry
gptkb:researchers
academia
gptkbp:usedFor gptkb:artificial_intelligence
gptkb:machine_learning
deep learning
gptkbp:website https://pytorch.org/
gptkbp:bfsParent gptkb:CUDA_Driver
gptkbp:bfsLayer 7