Descrição
Cargas JAX podem rodar por XLA em aceleradores AMD compatíveis com ROCm. Engenheiros de ML instalam esta variante jaxlib para treino e computação numérica com GPU. Versões de drivers, suporte de dispositivos, precisão e dados de modelos precisam de validação no hardware alvo.