Phase 5: JEPA + the Objective seam, integrated onto FSDP - #16

Closed

AshishKumar4 wants to merge 11 commits into

revamp/phase-3-parallelismfrom

revamp/integration

Commits on Jul 27, 2026

refactor(trainer): the Objective seam, one trainer for any learning problem

AshishKumar4
and
claude
committed
fb03cf1
feat(schedulers): flow matching, min-SNR-gamma weighting, EDM2 defaults

AshishKumar4
and
claude
committed
06887ce
feat(samplers): interval-limited classifier-free guidance

AshishKumar4
and
claude
committed
8ad7ac9
feat(metrics): FID on the vendored InceptionV3

AshishKumar4
and
claude
committed
aadc659
feat(autoencoder): per-dataset latent normalization on the shared seam

AshishKumar4
and
claude
committed
29d1a2d
feat(training): expose the flow matching, min-SNR and FID options

AshishKumar4
and
claude
committed
40f69d7
feat(jepa): I-JEPA and V-JEPA on the shared DiT blocks

AshishKumar4
and
claude
committed
f21bdd6
feat(jepa): finish the objective wiring and widen test coverage

AshishKumar4
and
claude
committed
033ef3b
merge revamp/phase-3-parallelism
AshishKumar4
committed
46e4d99
merge revamp/phase-5-jepa

AshishKumar4
and
claude
committed
2f498e0
test(jepa): prove the objective seam holds under FSDP and mixed precision

AshishKumar4
and
claude
committed
bc9d06b