Top #jax Tools & Software

Explore 3 hand-picked tools and software tagged with jax — ranked by popularity and community signals.

Popular Trending Recent Name A-Z

keras

github

Deep Learning for humans

AI Tools Python

★ 64,008

jax

github

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Developer Tools Python

★ 35,401

d2l-en

github

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

AI Tools Python

★ 28,623