I tried a lot of things but was not able to run Wandb on TPU.
Kernel
If there is something wrong in my code structure please let me know . Have raised an issue on github as well.
Cheers!
Thanks Harveen!
The conversation is continued Here