ChatGPT - PyTorch ML Example