A top artificial intelligence expert in the Obama, Trump and Biden administrations is leaving government.