Descrição
Cargas JAX podem executar por XLA em GPUs NVIDIA compatíveis com CUDA. Engenheiros de ML usam este runtime em treino acelerado, inferência e pesquisa numérica. Versões CUDA, drivers, uso de memória, precisão e tratamento de datasets precisam de validação nas máquinas alvo.