jax

b1cb3a5d - factor out process_map / post_process_map (#2788)

factor out process_map / post_process_map (#2788) * factor out process_map / post_process_map Also fix a bug from reusing post_process_call for pmap. Fixes #2787 * consolidate call_bind / map_bind code