Descrição
JAX pode usar um pacote binário pré-compilado do jaxlib para execução numérica baseada em XLA. Engenheiros de ML instalam para evitar compilação local ao rodar cargas de arrays e modelos. Compatibilidade binária, suporte a aceleradores e tratamento de dados de modelos ainda precisam de revisão.