Skip to main content
TrustRadius
JAX

JAX

Overview

What is JAX?

JAX, an open-source Google project, is Autograd and XLA, brought together for high-performance machine learning research. With its updated version of Autograd, JAX can automatically differentiate native Python and NumPy functions. It is free and open source under an Apache…

Read more
Recent Reviews
TrustRadius

Leaving a review helps other professionals like you evaluate Machine Learning Tools

Be the first one in your network to review JAX, and make your voice heard!

Return to navigation

Pricing

View all pricing

What is JAX?

JAX, an open-source Google project, is Autograd and XLA, brought together for high-performance machine learning research. With its updated version of Autograd, JAX can automatically differentiate native Python and NumPy functions. It is free and open source under an Apache 2.0 license.

Entry-level set up fee?

  • No setup fee

Offerings

  • Free Trial
  • Free/Freemium Version
  • Premium Consulting/Integration Services

Would you like us to let the vendor know that you want pricing?

Alternatives Pricing

What is Hugging Face?

Hugging Face is an open-source provider of natural language processing (NLP) technologies.

What is Qlik Sense?

Qlik Sense® is a self-service BI platform for data discovery and visualization. It supports a full range of analytics use cases—data governance, pixel-perfect reporting, and collaboration. Its Associative Engine indexes and connects relationships between data points for creating actionable insights.

Return to navigation

Product Details

What is JAX?

JAX, an open-source Google project, is Autograd and XLA, brought together for high-performance machine learning research. With its updated version of Autograd, JAX can automatically differentiate native Python and NumPy functions. It can differentiate through loops, branches, recursion, and closures, and it can take derivatives of derivatives of derivatives. It supports reverse-mode differentiation (a.k.a. backpropagation) via grad as well as forward-mode differentiation, and the two can be composed arbitrarily to any order.

JAX is free and open source under an Apache 2.0 license.

JAX Technical Details

Operating SystemsUnspecified
Mobile ApplicationNo
Return to navigation

Comparisons

View all alternatives
Return to navigation

Reviews

Sorry, no reviews are available for this product yet

Return to navigation