eduardoleao052/js-pytorch
A JavaScript implementation of PyTorch providing automatic differentiation, tensor operations, and GPU-accelerated neural network building blocks.

Velocity · 7d
+1.5
★ / day
Trend
→steady
star history
This repository implements a deep learning framework in JavaScript that closely mirrors PyTorch’s API and syntax. It provides tensor data structures and operations, automatic differentiation for gradient computation, and GPU acceleration through GPU.js. Users can build and train neural network models directly in JavaScript for browser or Node.js environments.