AMD Primus training framework source and docs

Primus on GitHub

<h2 id="amd-primus">AMD Primus</h2>
<p>This 1-Click image ships <strong>AMD Primus</strong>, AMD&#39;s high-performance training framework for large-scale foundation-model training on AMD Instinct GPUs (MI300X). It bakes in the AMD-published Primus training container (<code>rocm/primus:v26.4</code>) with the full ROCm stack (PyTorch, TransformerEngine, AITER) and the Megatron-LM and TorchTitan backends, plus the Primus repository with submodules pinned to a release tag.</p>
<p>Everything is baked into the snapshot at build time, so the droplet is instant-on: no multi-gigabyte image pull or submodule clone at first use.</p>
<h3 id="two-ways-to-use-it">Two ways to use it</h3>
<ol>
<li><strong>JupyterLab notebook</strong> on port 80 (fresh per-droplet token), with a guided quickstart notebook.</li>
<li><strong>primus-cli</strong> inside the container (<code>docker exec -it rocm /bin/bash</code>) for a real Llama-2 7B mock-data pretraining example (single-GPU-ready).</li>
</ol>
<p>See the Primus documentation for backends, configuration, and multi-node training.</p>

Primus

ROCm

PyTorch

JupyterLab

Docker

AMD Primus training framework on Ubuntu 24.04, ready to pretrain and post-train LLMs on AMD Instinct GPUs with Megatron-LM and TorchTitan backends.

AMD Developer Cloud

AMD Primus™

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

Secure your network with a VPN, protect your database, and more, with this selection of 1-Click Apps.

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 AMD Primus™