Ai tävlade för Kina vid olympiska sommarspelen 2016 i Rio de Janeiro, där hon blev utslagen i semifinalen på 200 meter frisim. Ai tävlade även i Kinas lag som slutade på fjärde plats på 4 × 200 meter frisim.