Sequential Morris Water Maze (sWM) This repository is for 2024 ICML Paper: Rapid Learning without Catastrophic Forgetting in the Morris Water Maze