Thank you for replying
Yep, I have tried installing NumPy both before and after installing PyTorch XLA.
I believe the error rises from a NumPy version mismatch.
I had missed a warning which essentially said - Compiled with numpy version 0xe but found 0xd.
The problem is that  pip install --upgrade numpy does not work because Kaggle loads a few modules (like Numpy and Pandas) . So one has to reload the module.
I did that too but it didn’t work.
If someone has successfully done it please let me know.
PS - Sorry this was not a pytorch specific question. I had’nt thought of numpy mismatch and environment pre-loading