Basic Pytorch Example

This is a Devbox for running PyTorch.

Note: This demo currently only works on WSL3 with Nvidia GPUs. For Linux, additional packages and configuration may be needed.