(draft) feat: add LeWM, an action-conditioned latent world model trained with SIGReg - #2032

Draft

gabrielfruet wants to merge 16 commits into

Commits on Aug 15, 2026

Add LeWM latent world model

gabrielfruet
and
claude
committed
0a26e73
Reach the fused attention kernel through getattr

gabrielfruet
and
claude
committed
9021781

Commits on Aug 28, 2026

refactor: require fused attention in LatentDynamicsPredictor

gabrielfruet
and
claude
committed
fea0671
docs: document keyword-only forward and ONNX export in LeWMLoss

gabrielfruet
and
claude
committed
44af827
feat: add LeWMProjectionHead to the head api

gabrielfruet
and
claude
committed
810f111
refactor: parameterize LeWM example and use LeWMProjectionHead

gabrielfruet
and
claude
committed
4eea61f
Merge branch 'master' into feat/lewm
gabrielfruet
authored
642c432
refactor: remove unnecessary forward comment in LeWMLoss

gabrielfruet
and
claude
committed
55ee663
refactor: rely on LeWM example module defaults instead of constants

gabrielfruet
and
claude
committed
9819388

Commits on Aug 31, 2026

Merge branch 'master' into feat/lewm
gabrielfruet
authored
96e063e

Commits on Sep 1, 2026

feat: make LeWM predictor conditioning optional and expose PredictorBlock

gabrielfruet
and
claude
committed
f9e4a51
feat: add batch_norm opt-out to LeWMProjectionHead

gabrielfruet
and
claude
committed
5b088b0
docs: scope LeWM loss docstrings to continuous-latent methods

gabrielfruet
and
claude
committed
4ec5312
fix: validate LeWM loss embeddings and guard rollout output_dim

gabrielfruet
and
claude
committed
b640f07
docs: fix LeWM example shape contract, timm scope and run command

gabrielfruet
and
claude
committed
60c1b22
refactor: extract AdaLNZero conditioning module in LeWM predictor

gabrielfruet
and
claude
committed
623af45