I was trying to replicate this model, which uses MDLSTM cells (whose formulation I found here). However, I’m struggling to find any implementation and trying to bring this formulation into a custom layer of the PyTorch framework. I also found a thread in this forum about the topic, but didn’t get much from it.
So, I would like to ask if anyone knows if there is any implementation of the MDLSTM networks in this framework or knows how to implement their formulation.
Thanks in advance.