https://www.mukeshmithrakumar.com/blog/deep-learning-with-tf2-numerical-computation/