lead ai engineer Jobs in california