Expose nd.rolling_apply function