JAX

Overview

High-performance array computing and autodiff — research favorite for large-scale models.

JAX is listed in Axioniq’s curated Open Source AI collection. Open-source models, frameworks, and communities that keep the AI ecosystem inspectable and portable.

At a glance

• Listed under Open Source AI.
• Composable numerical computing
• Tagged for research.
• Tagged for autodiff.
• Tagged for Python.

Who it’s for

• operators researching AI and enterprise technology vendors

Category context

Open-source models, frameworks, and communities that keep the AI ecosystem inspectable and portable.

Compare peers in Open Source AI or jump straight to the official JAX site.