Revisit TensorFlow specific docstrings #4
Description
A lot of docstrings are TensorFlow specific, like references to tf.function and similar in various code examples, but it maybe a lot of effort to update them all.
Originally posted by @dime10 in https://github.com/PennyLaneAI/diastatic-malt/pull/2#discussion_r1476293558