Thanks for the heads up about graphQL—I think I'll avoid for now. It will be for a new project so not a big deal.
I just had a quick look at PyTorch... you're totally right this is miles ahead of what I remember.
I forgot to ask about package management. Do you use pip, conda, poetry?