Add embedding and tiling diagnostics evals

[Hgherzog]

Summary

• Add embedding geometry diagnostics and tiling artifact metrics for eval-time representation checks.
• Extend pretrain-subset eval adapters so diagnostics can run on fixed-size pretraining samples and optional pinned subsets.
• Wire diagnostics through downstream evaluator logging and document the workflows.

Test plan

• uv run pytest -vv tests/unit/eval/test_embedding_diagnostics.py tests/unit/eval/datasets/test_pretrain_subset.py
• uv run python - <<'PY'

  from olmoearth_pretrain.evals.datasets.configs import dataset_to_config, get_eval_mode
  from olmoearth_pretrain.internal.all_evals import TILING_DIAG_TASKS
  print(get_eval_mode(dataset_to_config('pretrain_subset').task_type))
  print(get_eval_mode(dataset_to_config('pretrain_subset_128').task_type))
  print(sorted(TILING_DIAG_TASKS))

  PY

Notes

• Split out from the original combined draft PR Add RoPE and temporal RoPE experiments #571.

Made with Cursor