jax vs tfjs
jax and tfjs share 3 technologies: jaxlib, opt-einsum, and ml-dtypes. jax is written in Python and tfjs in TypeScript.
What they share and what sets them apart
jax-ml
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python36.6kRead on November 6, 2026, at commit 88d646cWhat is jax built with?In common
- jaxlib
- opt-einsum
- ml-dtypes
tensorflow
tfjs
A WebGL accelerated JavaScript library for training and deploying ML models.
TypeScript19.2kRead on November 6, 2026, at commit 7f5309fWhat is tfjs built with?Only in tfjs
- tensorboard-data-server
h5py
werkzeug
- React
@react-native-async-storage/async-storage
react-native
- Preact
prompt-toolkit
- Rich
- Keras
- optax
- orbax-checkpoint
- tensorflow-io-gcs-filesystem
- TensorFlow
tf-keras
flax
jax
- tensorflow-hub
- markdown
markupsafe
- progress
- expo-gl
- expo-camera
- jpeg-js
- glamor
vega-embed
d3-selection
vega-lite
Side by side
jaxtfjs
SizeLines written by people and agents, without generated ones; code files, without tests.
jax656,473 lines written601 code files
tfjs383,524 lines written2,527 code files
Who writes itA commit counts as AI's when it carries its signature: a Co-Authored-By line, its author name or the merge of its branch.
jax0% of its commits signed by AIWith Gemini CLI
tfjs0% of its commits signed by AI
ActivityFrom GitHub and its history, on the day SeeRepo read it.
jaxLast push November 6, 202630,226 commits36,582 stars
tfjsLast push June 23, 20266,042 commits19,155 stars
TestsThe files that test the code, and the share of code files a test imports.
jax391 test files52.2% of its code files have a test that imports them
tfjs72 test files1.9% of its code files have a test that imports them
Frequently asked questions
Which is bigger, jax or tfjs?
jax, with 656,473 lines written; tfjs has 383,524.
The proof: jax-ml/jax's code at commit 88d646c
Which has more GitHub stars, jax or tfjs?
jax, with 36,582 stars; tfjs has 19,155.
The proof: jax-ml/jax on GitHub
Which has more tests, jax or tfjs?
jax, with 391 test files; tfjs has 72.
The proof: jax-ml/jax's code at commit 88d646c
Which was updated more recently, jax or tfjs?
jax: its last push was on November 6, 2026, and tfjs's on June 23, 2026.
The proof: jax-ml/jax's commits on GitHub
See inside your code, today.
Paste your repo and see it drawn, part by part: a medium one in under a minute.
Free, no account.

