Vibe Coding Discoverpytorch
View on GitHubTensors and Dynamic neural networks in Python with strong GPU acceleration
PyTorch is a Python deep-learning framework for tensor computation, automatic differentiation, and dynamic neural networks. It supports CPU and GPU execution, including CUDA and ROCm acceleration.
Use Cases
Train neural networksRun GPU-accelerated tensor computationsBuild dynamic neural network architecturesCompute gradients with automatic differentiationPrototype deep learning researchDeploy models on CPU and GPU
Built With
- Language
- Python
- Frameworks
- CUDA · ROCm · cuDNN · NCCL · oneDNN
Tags
deep learning · tensor library · autograd · GPU acceleration · neural networks · CUDA · ROCm · scientific computing