jax.extend.linear_util.merge_linear_aux

jax.extend.linear_util.merge_linear_aux(aux1, aux2)