(*args, **kwargs)
return backend()['pmap'](*args, **kwargs)
Parallel-map to apply a function on multiple accelerators in parallel.
Parallel-map to apply a function on multiple accelerators in parallel.
Parallel - map to apply a function on multiple accelerators in parallel .
def pmap(*args, **kwargs):
  """Parallel-map to apply a function on multiple accelerators in parallel."""
  return backend()['pmap'](*args, **kwargs)
def pmap ( * args , * * kwargs ) : return backend ( ) [ 'pmap' ] ( * args , * * kwargs )