There is a function in my code base that is "already vmapped", i.e. when fed an array of shap