← all repositories

eduardoleao052/js-pytorch

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

1.2k stars JavaScript ML Frameworks
js-pytorch
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.