pytorch/pytorch
PyTorch
PyTorch is an open framework for tensor computation and deep learning in Python, with dynamic neural networks and GPU support.
Catalog projects marked with #gpu. Tags work as dedicated landing pages, so related tools are easier to find and connect.
This collection holds 2 projects with a combined 105,344 GitHub stars. Main languages: Python, Julia.
PyTorch is an open framework for tensor computation and deep learning in Python, with dynamic neural networks and GPU support.
Flux.jl is a machine-learning library for Julia with lightweight abstractions over GPU and automatic differentiation support.