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

JAX Setup

Note

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

JAX setup on Ubuntu 22

Ubuntu 22 (Ubuntu22 AMI)

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