This reminds me of one I read a few years ago “fizzbuzz in tensorflow”

https://joelgrus.com/2016/05/23/fizz-buzz-in-tensorflow/