Computational Lithography to Enable Faster AI Development