I came across a PYPI package called spark_tensorflow_distributor with an example on running a tensorflow job on a spark cluster.
To create the Tensorflow job on a spark c