Simplify infer API