xla

f43941b7 - Do not use TLS for replication devices as calls into them from autograd are coming from "alien" threads.