Yes, an O-1 visa might be a viable alternative for you, given your extensive research in AI/ML and publications. The O-1 visa is designed for individuals with extraordinary ability in their field, which includes sciences, technology, engineering, and mathematics (STEM) fields like AI/ML.

To strengthen your O-1 petition, focus on demonstrating your extraordinary ability through evidence such as publication in top-tier journals or conferences, presentations or keynote speeches at prominent events, etc