JS-PyTorch
A deep learning JavaScript library built from scratch with PyTorch-like syntax and GPU acceleration via GPU.js.
What is JS-PyTorch?
JS-PyTorch is a deep learning library for JavaScript that replicates PyTorch's syntax and functionality. It allows developers to build, train, and deploy neural networks entirely in JavaScript, with GPU acceleration for performance. The library solves the problem of running machine learning models directly in web browsers or Node.js without relying on Python backends.
JavaScript developers and researchers who want to implement deep learning models in web applications or Node.js environments, especially those familiar with PyTorch's API.
Developers choose JS-PyTorch for its close adherence to PyTorch's syntax, making it easy to port models from Python, and its built-in GPU acceleration which enables performant training and inference directly in JavaScript.
Overview
A JavaScript library like PyTorch, with GPU acceleration.
Use Cases
Best For
- Building and training neural networks directly in the browser
- Porting PyTorch models to JavaScript applications
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
