tfjs and TSTorch
These are competitors: TSTorch is an educational reimplementation of PyTorch concepts for WebGPU environments, while TensorFlow.js is the established production-grade library for the same use case of browser-based ML model training and inference.
About tfjs
tensorflow/tfjs
A WebGL accelerated JavaScript library for training and deploying ML models.
Supports both browser and Node.js execution through pluggable backends (WebGL, WebAssembly, WebGPU, and native TensorFlow C++), enabling client-side inference and retraining with sensor data. Provides layered APIs from low-level linear algebra operations to high-level Keras-compatible layers, plus tools to import pre-trained TensorFlow and Keras models. Includes specialized packages for data processing, visualization, and AutoML model loading, with individual module imports available to minimize bundle size.
About TSTorch
r-chong/TSTorch
A PyTorch-style runtime library in TypeScript + WebGPU. Built to understand how ML frameworks and models work internally.
Related comparisons
Scores updated daily from GitHub, PyPI, and npm data. How scores work