The federal government wants to revamp the education system with AI.