Descrição
Transformações JAX podem executar pelo runtime XLA fornecido pelo jaxlib. Desenvolvedores e pesquisadores usam em computação numérica acelerada, autodiff e cargas de aprendizado de máquina. Compatibilidade do runtime, precisão e uso de recursos devem ser validados em cada plataforma.