Function mlx_value_and_grad 

pub unsafe extern "C" fn mlx_value_and_grad(
    res: *mut mlx_closure_value_and_grad,
    fun: mlx_closure,
    argnums: *const c_int,
    argnums_num: usize,
) -> c_int