<p>Run a ready-to-use Docker image with JAX 0.11.1 and JupyterLab on top of your ROCm-enabled AMD Instinct system. JAX is installed with AMD&#39;s official ROCm 7.14 PJRT plugin, so jax.jit compiles and executes directly on the GPU. All software is preinstalled in containers, so you can start straight from either SSH or Jupyter.</p>

ROCm (host system)

jaxlib

ROCm PJRT plugin (jax_rocm7_pjrt, jax_rocm7_plugin)

JupyterLab (in Docker image)

JAX 0.11.1 with the ROCm 7.14 PJRT plugin for AMD Instinct GPUs. Autograd, JIT and XLA, with JupyterLab.

AMD Developer Cloud

Focus on building great things, while you use these 1-Click Apps to build your business.

These 1-Click Apps make it simple to deploy and manage your clients' infrastructure. Save time and resources, while reducing headaches and passing the value onto your clients.

Apps for agencies and website development shops

Work more effectively from home or remotely with these 1-Click Apps.

Work more effectively from home or remotely

Stop spending hours editing YAML and Helm charts. Instead, use these popular, preconfigured solutions to add apps to your Kubernetes cluster in minutes instead of days.

1-Click Apps for managing Kubernetes environments

Stay focused on teaching and learning, instead of managing infrastructure. 

1-Click Apps for students and educational institutions

SaaS Add-Ons make your DigitalOcean infrastructure better! Check out the full list of SaaS Add-Ons here!  

New Feature Alert: Introducing Software License Subscriptions on DigitalOcean Marketplace!

We now support Droplet 1-Click updates via API. See documentation for the details

Check out Tabnine, the AI coding assistant that accelerates & simplifies software development.

Description

Software Included

Getting started after deploying JAX