AI lithography control