add pipeline stage inference functionality