Action Jets: Learning the Local Shape of the Action-conditioned Data Manifold

World models are a powerful means of reasoning and planning, but many of these action-conditioned predictive models are high dimensional and computationally expensive. These challenges limit the number of action candidates that can be evaluated by a planner.