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.
TensorFlow.js helps web developers build and run machine learning models directly in web browsers or Node.js applications. It takes raw data, pre-trained TensorFlow or Keras models, and outputs interactive ML experiences without server-side processing. This allows front-end developers and web application builders to integrate AI capabilities into their projects.
About TSTorch
r-chong/TSTorch
A PyTorch-style runtime library in TypeScript + WebGPU. Built to understand how ML frameworks and models work internally.
This is a machine learning library built from scratch in TypeScript, designed for those who want to understand the inner workings of ML frameworks. You can input data like numbers or arrays and define neural network structures using layers and activation functions. The output includes trained models capable of making predictions and insights into how automatic differentiation and tensor operations function. This tool is ideal for developers interested in learning about machine learning framework internals or building custom ML components in TypeScript.
Related comparisons
Scores updated daily from GitHub, PyPI, and npm data. How scores work