mstar.model.vjepa2.components

mstar.model.vjepa2.components#

Modules

ac_predictor

V-JEPA 2 action-conditioned predictor (for V-JEPA 2-AC).

layers

Shared transformer building blocks for V-JEPA 2 encoder and predictor.

predictor

V-JEPA 2 masked latent predictor.

rope_utils

3D rotary position embedding helpers for V-JEPA 2.

vit_encoder

V-JEPA 2 video encoder (ViT with 3D tubelet patches + 3D RoPE).