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
JAX 0.8.2 on Ubuntu 24.04 with free maintenance support by PCloudhosting. This is a repackaged open-source software product, for which additional support charges apply./strong>. This is a nicely repackaged open-source product premium support is available if you ever need extra help (additional charges may apply).
JAX is a fantastic open-source library from Google Research that's perfect for machine learning and high-performance numerical computing. It feels just like good old NumPy, but it packs some really powerful features under the hood:
- Automatic differentiation (autodiff) super easy gradients for anything you write
- Automatic vectorization (vmap) apply functions across batches without loops
- Just-in-time (JIT) compilation with XLA get massive speedups on accelerators
- Parallelization (pmap) spread work across multiple devices effortlessly
Why People Are Loving JAX
- Drop-in NumPy replacement most of your existing code just works, but now it's faster + you get free gradients
- Zero-to-hero hardware support run the same code on CPU, GPU, or TPU with almost no changes
- Functional style + composable transforms = clean, reusable, and blazing-fast code
- Powers some of the biggest models out there (Gemini, Gemma, and tons of research projects)
- Huge and growing ecosystem: Flax, Optax, Orbax, JAX-Privacy, Pallas for custom kernels, and more
Key Highlights
- Always up-to-date & secureBitnami watches for vulnerabilities and new releases, then automatically repackages and pushes fresh versions
- Lightweight & efficient built on minimal base images following best practices starts fast and stays lean
- Easy to configure rolling version tags let you pick exactly what you want and run it anywhere
- Optimized for AWS Cloud Marketplace deploy confidently on EKS, EC2, or other AWS services with secure, ready-to-go packaging
Whether you're just experimenting with new ideas, training huge models on GPUs/TPUs, doing serious scientific simulations, or want NumPy on serious performance steroids JAX makes high-performance computing feel surprisingly approachable and fun.
Highlights
- Acts as a drop-in replacement for NumPy with added performance and autodiff capabilities.
- Runs seamlessly on CPUs, GPUs, and TPUs with minimal code changes.
- Acts as a drop-in replacement for NumPy with added performance and autodiff capabilities.
Details
Sold by
Delivery method
Delivery option
64-bit (x86) Amazon Machine Image (AMI)
Latest version
Operating system
Ubuntu 24.04
Deployed on AWS
New
Introducing multi-product solutions
You can now purchase comprehensive solutions tailored to use cases and industries.
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.
If you are an AWS Free Tier customer with a free plan, you are eligible to subscribe to this offer. You can use free credits to cover the cost of eligible AWS infrastructure. See AWS Free Tier for more details. If you created an AWS account before July 15th, 2025, and qualify for the Legacy AWS Free Tier, Amazon EC2 charges for Micro instances are free for up to 750 hours per month. See Legacy AWS Free Tier for more details.
Dimension | Cost/hour |
|---|---|
m4.large Recommended | $0.008 |
t2.micro | $0.001 |
t3.micro | $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
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.
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.
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).
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 AWS ParallelCluster, with over 100 HPC and AI/ML tools, a remote desktop environment (DCV), 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 TensorFlow, PyTorch, JAX, Keras, and Horovod tuned for AWS ParallelCluster, 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 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.