The US can compete with China in AI education -- here's how