tf.keras.layers.TimeDistributed equivalent in PyTorch

[ptrblck]

This topic seems to have a few implementations.