AI hardware resource allocation