Before revision:  It dynamically samples from a flexible number of distinct SR maps while accumulating evidence about the current reward context , and outperforms competing algorithms  in settings with both known and unsignalled rewards changes.
After revision:  It dynamically samples from a flexible number of distinct SR maps while accumulating evidence about the current task context through amortized inference. It improves SR's transfer capabilities and outperforms competing algorithms in settings with both known and unsignalled rewards changes.