Function expand_with 

pub fn expand_with<A: CubeIndexExpand<Output = ExpandElementTyped<V>>, V: CubeType>(
    scope: &mut Scope,
    expand: A,
    index: A::Idx,
) -> ExpandElementTyped<V>