lead machine learning engineer Jobs in california