(*args, **kwargs)
return backend()['psum'](*args, **kwargs)
Parallel-sum to use within a pmap'd function for aggregation.
Parallel-sum to use within a pmap'd function for aggregation.
Parallel - sum to use within a pmap d function for aggregation .
def psum(*args, **kwargs):
  """Parallel-sum to use within a pmap'd function for aggregation."""
  return backend()['psum'](*args, **kwargs)
def psum ( * args , * * kwargs ) : return backend ( ) [ 'psum' ] ( * args , * * kwargs )