* override dist backend when using tpus * added test * updated doc string * drop redundant info... * more redundant info Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>