This is your work, valued
MotivationModels. Pytorch based library containing reinforcement learning agents with forward models and intrinsic motivation modules