Sold by: pCloudHosting LLCÂ
Deployed on AWS
AWS Free Tier
This product has charges associated with it for seller support. JAX is a high performance numerical computing library from Google that combines NumPy like syntax with automatic differentiation (autograd) and accelerated execution on CPUs, GPUs, and TPUs.
Overview
This is a repackaged open source software product wherein additional charges apply for support. JAX is an open source machine learning and numerical computing library developed by Google Research. It provides a familiar NumPy style API while offering advanced features such as automatic differentiation, vectorization, and just in time (JIT) compilation powered by XLA (Accelerated Linear Algebra).
Highlights
- High-performance computing
- NumPy-compatible API
- Automatic differentiation
Details
Sold by
Delivery method
Delivery option
64-bit (x86) Amazon Machine Image (AMI)
Latest version
Operating system
Ubuntu 24.04
Deployed on AWS
Unlock automation with AI agent solutions
Fast-track AI initiatives with agents, tools, and solutions from AWS Partners.
Features and programs
Financing for AWS Marketplace purchases
AWS Marketplace now accepts line of credit payments through the PNC Vendor Finance program. This program is available to select AWS customers in the US, excluding NV, NC, ND, TN, & VT.
Pricing
Pricing is based on actual usage, with charges varying according to how much you consume. Subscriptions have no end date and may be canceled any time.
Additional AWS infrastructure costs may apply. Use the AWS Pricing Calculator to estimate your infrastructure costs.
Dimension | Cost/hour |
|---|---|
m4.large Recommended | $0.008 |
t2.micro AWS Free Tier | $0.001 |
t3.micro AWS Free Tier | $0.008 |
m3.large | $0.008 |
t2.xlarge | $0.008 |
r5.large | $0.008 |
t2.small | $0.008 |
m5.large | $0.008 |
t3.small | $0.008 |
c4.large | $0.008 |
Vendor refund policy
No Refund
Legal
Vendor terms and conditions
Upon subscribing to this product, you must acknowledge and agree to the terms and conditions outlined in the vendor's End User License Agreement (EULA)Â .
Content disclaimer
Vendors are responsible for their product descriptions and other product content. AWS does not warrant that vendors' product descriptions or other product content are accurate, complete, reliable, current, or error-free.
Delivery details
64-bit (x86) Amazon Machine Image (AMI)
Amazon Machine Image (AMI)
An AMI is a virtual image that provides the information required to launch an instance. Amazon EC2 (Elastic Compute Cloud) instances are virtual servers on which you can run your applications and workloads, offering varying combinations of CPU, memory, storage, and networking resources. You can launch as many instances from as many different AMIs as you need.
Version release notes
Packaged with latest updates as of August/2025
Additional details
Usage instructions
Connect your instance via SSH, the username is ubuntu. More info on SSH: https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/AccessingInstancesLinux.html - Run the following commands: #sudo su #sudo apt update #pip show jax jaxlib
Support
Vendor support
Feel free to reach out anytime. Our support team is available 24x7 for assistance. Email: anant.shahi@pcloudhostings.com Website:
AWS infrastructure support
AWS Support is a one-on-one, fast-response support channel that is staffed 24x7x365 with experienced and technical support engineers. The service helps customers of all sizes and technical abilities to successfully utilize the products and features provided by Amazon Web Services.
Similar products
Bitnami Container Solution for JAX
By Bitnami by VMware
Up-to-date, customizable, and secure. JAX is a Python-based toolset (Autograd and XLA) for high performance machine learning applications. Features familiar API, transformations and multiple backend support.
Bitnami package for WildFly
By Bitnami by VMware
Up-to-date and secure image. Wildfly is a lightweight, open source application server, formerly known as JBoss, that implements the latest enterprise Java standards.
ParaTools Pro for E4S™: AI/ML & HPC Tools on ODDC Node (x86)
By ParaTools Inc.
This product has charges associated with it for optional seller support and pre-configured stack (ParaTools Pro for E4S™). ParaTools Pro for E4S™ is the hardened Extreme-scale Scientific Software Stack using HPC Cloud ODDC, with over 100 HPC and AI/ML tools, a remote desktop environment (VNC), and cluster configuration built using the Spack package manager and a proprietary MVAPICH MPI tuned for AWS Elastic Fabric Adapter (EFA). It is a platform for developing AI/ML applications using tools such as NVIDIA NeMo™, TensorFlow, PyTorch, JAX, Keras, and Horovod tuned for AWS and ODDC, with a complete HPC software stack including numerical libraries (such as WRF, PETSc, Trilinos, SuperLU-dist), visualization tools (such as ParaView and VisIt), performance evaluation tools (such as TAU and HPCToolkit), HPC applications (such as OpenFOAM, LAMMPS, Xyce, CP2K, deal.II, GROMACS and Quantum Espresso).
ParaTools Pro for E4S™: AI/ML & HPC Tools on ODDC Server (x86)
By ParaTools Inc.
This product has charges associated with it for optional seller support and pre-configured stack (ParaTools Pro for E4S™). ParaTools Pro for E4S™ is the hardened Extreme-scale Scientific Software Stack using HPC Cloud ODDC, with over 100 HPC and AI/ML tools, a remote desktop environment (VNC), and cluster configuration built using the Spack package manager and a proprietary MVAPICH MPI tuned for AWS Elastic Fabric Adapter (EFA). It is a platform for developing AI/ML applications using tools such as NVIDIA NeMo™, TensorFlow, PyTorch, JAX, Keras, and Horovod tuned for AWS and ODDC, with a complete HPC software stack including numerical libraries (such as WRF, PETSc, Trilinos, SuperLU-dist), visualization tools (such as ParaView and VisIt), performance evaluation tools (such as TAU and HPCToolkit), HPC applications (such as OpenFOAM, LAMMPS, Xyce, CP2K, deal.II, GROMACS and Quantum Espresso).
Customer reviews
No customer reviews yet
Be the first to review this product . We've partnered with PeerSpot to gather customer feedback. You can share your experience by writing or recording a review, or scheduling a call with a PeerSpot analyst.