For research purposes, I started looking into
tf.autodiff.ForwardProp. However, I have also noticed that supporting batches of tangents inside the function are not yet fully implemented. I saw that there has been some awesome work on this topic from a GSOC student last year, and I would like to bring it to an end or bring it forward at least!
However, it is worth noting that I am also a student, and I would need some support on the matter.