mstar.model.vjepa2.components

mstar.model.vjepa2.components#

Modules

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).