Any PyTorch function can work as Keras' Timedistributed?

The repo is not longer available.