def METHOD_NAME(name=None):
  """An op which initializes MPI on the device on which it is run.
  All future MPI ops must be run on the same device that the `init` op was run
  on.
  """
  return MPI_LIB.mpi_init(name=name)