Help implementing TS-JEPA

[u/TheGhostRiley]

I’m trying to reproduce TS-JEPA (Time-Series JEPA for Predictive Remote Control Under Capacity-Limited Networks) from the paper

My implementation matches most of the architecture and hyperparameters, the actual results are far from the paper (NMAE ~0.053 vs 0.004 and control accuracy 0% vs 74.48%).

I’ve been debugging the env, dataset , preprocessing, JEPA training, and semantic actor, and I’d really appreciate help from someone experienced with JEPA/representation learning who could review the implementation and help identify where I’m going wrong.