Frameworkai-frameworksvN/A

JAX

by Google · open-source · Last verified 2026-03-20

JAX is a numerical computation library that combines automatic differentiation and XLA (Accelerated Linear Algebra) compilation to enable high-performance machine learning research. It is particularly well-suited for research involving custom models and algorithms, offering flexibility and speed.

https://github.com/google/jax ↗

B+

B+—Good

Adoption: B+Quality: A+Freshness: ACitations: AEngagement: B+

Specifications

License: Apache 2.0
Pricing: open-source
Capabilities: automatic differentiation, gpu acceleration, cpu acceleration, xla compilation, high-performance computing
Integrations
Use Cases: machine learning research, scientific computing, deep learning, custom model development
API Available: Yes
Tags: numerical-computation, automatic-differentiation, xla, machine-learning, python
Added: 2026-03-20
Completeness: 100%

Index Score

78.5

Adoption

Quality

Freshness

Citations

Engagement

Put AI to work for your business

Deploy this framework alongside autonomous AaaS agents that handle tasks end-to-end — no babysitting required.

Start Free Learn more about the agent pipeline →

Explore the full AI ecosystem on Agents as a Service