Descrição
Arrays JAX podem ser trocados por MPI com comunicação zero-copy em cargas numéricas distribuídas. Pesquisadores e desenvolvedores HPC usam mpi4jax em simulações, treino paralelo e computação ciente de aceleradores. Resultados dependem de configuração MPI, posicionamento de dispositivos, sincronização e sementes reproduzíveis.