https://github.com/chillymango/torch-webgpu
https://github.com/pytorch-labs/attention-gym
https://github.com/pytorch-labs/monarch