Active Inference is kinda "a joint learning of the state representation with the search algorithm"...or at least, related to the idea. I like framing it as a joint learning problem.