This document is relevant for: Inf1, Inf2, Trn1, Trn2

JAX Setup

Note

This setup guide is relevant for Inf2 & Trn1 / Trn1n / Trn2 instances.

JAX setup on Ubuntu 22

Ubuntu 22 (Ubuntu22 AMI)

This document is relevant for: Inf1, Inf2, Trn1, Trn2