This is a tensorflow implementation of Recurrent Independent Mechanisms (Goyal et al. 2019) by Fuyuan Li, original authorship of the paper is
Anirudh Goyal, Alex Lamb, Jordan Hoffmann, Shagun Sodhani, Sergey Levine, Yoshua Bengio, Bernhard Sch{\"o}lkopf
@article{goyal2019recurrent,
title={Recurrent independent mechanisms},
author={Goyal, Anirudh and Lamb, Alex and Hoffmann, Jordan and Sodhani, Shagun and Levine, Sergey and Bengio, Yoshua and Sch{\"o}lkopf, Bernhard},
journal={arXiv preprint arXiv:1909.10893},
year={2019}
}
The pytorch implementation can be found at author's github repo [https://github.com/anirudh9119/RIMs]