Function vmap_replace 

pub fn vmap_replace(
    inputs: &MlxArrayVector,
    traced_inputs: &MlxArrayVector,
    traced_outputs: &MlxArrayVector,
    input_axes: &[i32],
    output_axes: &[i32],
) -> Result<MlxArrayVector, MlxCError>

Replaces one traced array in a vector-of-arrays graph, the building block of the VMAP transform.

Errors

Returns the captured MLX-C description when the replacement fails.