feat(world-model): FutureState feature-reconstruction (JEPA) loss + optional action conditioning Feat/future state jepa#80

Merged

m-zain-khawaja merged 3 commits into

autowarefoundation:mainfrom

gcordova10:feat/future-state-jepa

Jun 22, 2026

Commits on Jun 22, 2026

feat(world-model): add FutureState feature-reconstruction (JEPA) loss and optional action conditioning
gcordova10
committed
fix(world-model): guard against zero-sum temporal weights in JEPA loss
gcordova10
committed
fix(world-model): avoid list/Tensor type clash in JEPA loss for mypy
gcordova10
committed