Pytorch layers: https://github.com/vtempest/ai-research-agent/blob/master/src/train/neural-net.js gpt layers: https://github.com/vtempest/ai-research-agent/blob/master/test/transformer.test.js