JEPA - Geometry of planning. Curvature Regularization. Gradient-based Optimization in World Models .