Add V-JEPA 2 (Meta FAIR) distributed training test case#1035

Open

paragao wants to merge 16 commits intomainfrom

feat/vjepa2-distributed-training

Commits on Mar 23, 2026

Add V-JEPA 2 distributed training test case for p5en (H200)
paragao
committed
Add V-JEPA 2.1 distributed training test case with image+video co-training
paragao
committed
Add B200 (Blackwell) benchmark sbatch scripts for V-JEPA 2 and 2.1
paragao
committed
Increase synthetic dataset default to 50K for accurate benchmarks
paragao
committed
Add nsys profiling scripts for V-JEPA 2 and 2.1 on B200
paragao
committed
Add optimized config for V-JEPA 2 with torch.compile and no activation checkpointing
paragao
committed

Commits on Mar 25, 2026

Remove incorrect MFU calculation from parse_benchmark.py and add --gpu_type flag
paragao
committed

Commits on Apr 15, 2026

Commits on Apr 28, 2026

Update 3.test_cases/pytorch/vjepa2.1/slurm/optim_sweep_b200.sbatch
KeitaW
authored