I think someone just needs to expose this in Python you could probably do so with
TVM_REGISTER_GLOBAL("SimplifyStmt").set_body_typed([](tir::Stmt stmt) {
arith::Analyzer analyzer;
return arith::StmtSimplifier(&analyzer).Simplify(stmt);
});
and in Python
tvm.get_global_func("SimplifyStmt")(my_stmt)