I believe this is what the author tried first in the post. He even links to this test UI where you can compare the "plain math" approach to the neural network:

https://attardi.org/pytorch-and-coreml-test-ui/