Yes. It's handy to see if you don't have any obvious bias in your data. I don't know how it works in LLMs though, you'll probably need to do some fancy tricks to visualize that.
I understand LLMs and ML models are not linear functions but, they are functions (no?) and some complex higher level math can be grasped intuitively when visualized.
[1] https://news.ycombinator.com/item?id=43495585
Edit: thoughts.