SeeRepo
Draw my repo

What is jax built with?

jax is a Python repo. It uses AI through jaxlib, opt-einsum, and ml-dtypes. It has 656,473 written lines, with 391 test files.

Read on November 6, 2026, at commit 88d646c. Read it as markdown

Backend

AI 3

  • jaxlib46 files
  • opt-einsum4 files
  • ml-dtypes2 files

Support

Other 2

  • NumPy218 files
  • SciPy4 files

Other repos with a similar stack

Ordered by how much of its stack they share with jax, then by stars.

Frequently asked questions

What language is jax written in?
jax is written in Python, with 656,473 written lines.
Does jax use AI?
Yes. jax uses AI through jaxlib, opt-einsum, and ml-dtypes.
Does jax have tests?
Yes. jax has 391 test files.
Which repos have a stack like jax's?
The ones that share most of its stack:tensorflow/tfjsray-project/raykeras-team/keras

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.

Or look inside SeeRepo's own →

Is it your repo? Add its badge, keep it up to date or take it down →