Starting at $0
View PricingOverview
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…
Recent Reviews
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!
Get StartedPricing
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
Product Details
- About
- Tech 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 is free and open source under an Apache 2.0 license.
JAX Technical Details
Operating Systems | Unspecified |
---|---|
Mobile Application | No |