Shuaijun commited on
Commit
f3d43a3
·
verified ·
1 Parent(s): 18c6f0e

Upload 38 files

Browse files
Files changed (38) hide show
  1. README.md +76 -1
  2. granular-multiview/README.md +13 -0
  3. granular-multiview/config.json +51 -0
  4. granular-multiview/model.pt +3 -0
  5. granular-relational/README.md +13 -0
  6. granular-relational/config.json +46 -0
  7. granular-relational/model.pt +3 -0
  8. manifest.json +702 -0
  9. pushobj-unseen-shapes/README.md +13 -0
  10. pushobj-unseen-shapes/config.json +36 -0
  11. pushobj-unseen-shapes/model.pt +3 -0
  12. pusht-adapted-predictor-full/README.md +11 -0
  13. pusht-adapted-predictor-full/config.json +73 -0
  14. pusht-adapted-predictor-full/model.pt +3 -0
  15. pusht-exact-realization-full/README.md +11 -0
  16. pusht-exact-realization-full/config.json +205 -0
  17. pusht-exact-realization-full/model.pt +3 -0
  18. pusht-multi-geometry/README.md +13 -0
  19. pusht-multi-geometry/config.json +76 -0
  20. pusht-multi-geometry/model.pt +3 -0
  21. pusht-predictive-adaptation/README.md +11 -0
  22. pusht-predictive-adaptation/config.json +75 -0
  23. pusht-predictive-adaptation/model.pt +3 -0
  24. pusht-relational/README.md +13 -0
  25. pusht-relational/config.json +26 -0
  26. pusht-relational/model.pt +3 -0
  27. pusht-visual-shifts/README.md +13 -0
  28. pusht-visual-shifts/config.json +36 -0
  29. pusht-visual-shifts/model.pt +3 -0
  30. reacher-relational/README.md +13 -0
  31. reacher-relational/config.json +23 -0
  32. reacher-relational/model.pt +3 -0
  33. reacher-temporal-transport/README.md +11 -0
  34. reacher-temporal-transport/config.json +24 -0
  35. reacher-temporal-transport/model.pt +3 -0
  36. reacher-world-model-full/README.md +11 -0
  37. reacher-world-model-full/config.json +27 -0
  38. reacher-world-model-full/model.pt +3 -0
README.md CHANGED
@@ -1,3 +1,78 @@
1
  ---
2
- license: apache-2.0
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language:
3
+ - en
4
+ tags:
5
+ - world-model
6
+ - jepa
7
+ - robotics
8
+ - decision-alignment
9
+ library_name: pytorch
10
+ datasets:
11
+ - Shuaijun/D-JEPA-Dataset
12
  ---
13
+
14
+ # D-JEPA checkpoints
15
+
16
+ **[Project website](https://nebulis-lab.com/D-JEPA)** ·
17
+ **[Model repository](https://huggingface.co/Shuaijun/D-JEPA)** ·
18
+ **[Decision-supervision dataset](https://huggingface.co/datasets/Shuaijun/D-JEPA-Dataset)** ·
19
+ **[Code](https://github.com/NEBULIS-Lab/D-JEPA)**
20
+
21
+ Task/module profiles for **D-JEPA: A Decision-Aligned Latent World Model**.
22
+ See each profile's `README.md` and `config.json` for architecture, provenance,
23
+ upstream dependencies, calibrated decisions and tensor-file checksums.
24
+
25
+ ## Choose a checkpoint profile
26
+
27
+ | Task | Profile | Contents |
28
+ |---|---|---|
29
+ | PushT | [Relational alignment](./pusht-relational/README.md) | Learned decision-alignment module |
30
+ | PushT | [Predictive adaptation](./pusht-predictive-adaptation/README.md) | Updated parameters; requires the exact base model |
31
+ | PushT | [Adapted predictor](./pusht-adapted-predictor-full/README.md) | Full predictor weights |
32
+ | PushT | [Multi-geometry alignment](./pusht-multi-geometry/README.md) | Module using four predictive geometries |
33
+ | PushT | [Exact representation realization](./pusht-exact-realization-full/README.md) | Two predictors and relation module in one tensor file |
34
+ | Reacher | [Relational alignment](./reacher-relational/README.md) | Task-specific decision-alignment module |
35
+ | Reacher | [Temporal transport](./reacher-temporal-transport/README.md) | Bounded future-update module |
36
+ | Reacher | [Full model](./reacher-world-model-full/README.md) | Predictors, alignment and transport weights |
37
+ | Granular | [Relational alignment](./granular-relational/README.md) | Spatial-feature alignment module |
38
+ | Granular | [Multi-view alignment](./granular-multiview/README.md) | Four-view variant using the same backbone |
39
+ | PushObj | [Unseen-shape transfer](./pushobj-unseen-shapes/README.md) | Task-local ordinal alignment module |
40
+ | PushT | [Visual-shift alignment](./pusht-visual-shifts/README.md) | Corruption-trained ordinal alignment module |
41
+
42
+ These are configurations of **D-JEPA**, not 12 separate methods. Each profile
43
+ contains `model.pt`, `config.json` and its own card. The [manifest](./manifest.json)
44
+ indexes source and exported weight hashes.
45
+
46
+ ## Loading and dependencies
47
+
48
+ All `model.pt` files contain tensor dictionaries and support
49
+ `torch.load(path, map_location="cpu", weights_only=True)`. Exact tensor values
50
+ from selected formal checkpoints are preserved. Packaging removes optimizer
51
+ state and private path metadata; exported file hashes are therefore distinct
52
+ from original training-file hashes.
53
+
54
+ ```python
55
+ import torch
56
+
57
+ state = torch.load("pusht-relational/model.pt", map_location="cpu", weights_only=True)
58
+ ```
59
+
60
+ This reads a tensor state dictionary, not an instantiated model. Use the matching
61
+ D-JEPA architecture and configuration; these custom modules are not loaded through
62
+ Transformers `AutoModel.from_pretrained`.
63
+
64
+ Profiles with `upstream_included: true` contain pretrained predictors. Other
65
+ profiles contain our learned modules or updated parameter subsets and require
66
+ their recorded predictive models/features. `pusht-exact-realization-full` is
67
+ one file containing two predictors and a relational operator, not a distilled
68
+ single-backbone student.
69
+
70
+ Reacher's physical result uses the relational selector; temporal transport is
71
+ also provided but its native-distance diagnostic is not a substitute result.
72
+ Multi-geometry's 128-start development result remains a separate protocol from
73
+ the independent 256-start PushT evaluation.
74
+
75
+ Local release candidate: authors have not yet selected publication licenses or
76
+ completed all upstream weight redistribution checks. This card grants no new
77
+ rights to third-party material. The author-designated repository is
78
+ `Shuaijun/D-JEPA`.
granular-multiview/README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: granular-multiview
2
+
3
+ - Architecture: `granular`.
4
+ - Tensor count: 30.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `e796fa3ee2fbc6f5b4abdfae99e8fa843606f923284a7fb95f59c0fc6c512c70`.
7
+ - Exported tensor-file SHA256: `2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Input width: 4. Gate: `base_minus_refined`, strict threshold `-0.010684727949480862`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
12
+
13
+ Project license is selected by the authors before public upload.
granular-multiview/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "granular-multiview",
4
+ "source_sha256": "e796fa3ee2fbc6f5b4abdfae99e8fa843606f923284a7fb95f59c0fc6c512c70",
5
+ "weights_sha256": "2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff",
6
+ "tensor_count": 30,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "granular",
9
+ "model": {
10
+ "input_dim": 4,
11
+ "hidden_dim": 64,
12
+ "low_rank": 8,
13
+ "max_correction": 0.2
14
+ },
15
+ "input_dim": 4,
16
+ "gate_threshold": -0.010684727949480862,
17
+ "gate_mode": "base_minus_refined",
18
+ "gate_decimals": null,
19
+ "fusion_weights": [
20
+ 0.0,
21
+ 0.35,
22
+ 0.6,
23
+ 0.05
24
+ ],
25
+ "upstream_included": false,
26
+ "lineage": {
27
+ "schema": "dajepa_granular_train_lineage_v1",
28
+ "method": "da_jepa_mg",
29
+ "dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
30
+ "config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
31
+ "adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
32
+ "train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
33
+ "calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
34
+ "train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
35
+ "calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
36
+ "trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
37
+ "core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
38
+ },
39
+ "training": {
40
+ "seed": 20260831,
41
+ "updates": 1500,
42
+ "batch_starts": 16,
43
+ "learning_rate": 0.0003,
44
+ "weight_decay": 0.0001,
45
+ "max_grad_norm": 1.0,
46
+ "tail_k": 16,
47
+ "evaluate_every": 25,
48
+ "mg_grid_step": 0.05,
49
+ "supervision_count": 16
50
+ }
51
+ }
granular-multiview/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff
3
+ size 279105
granular-relational/README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: granular-relational
2
+
3
+ - Architecture: `granular`.
4
+ - Tensor count: 30.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `8f613280f8084e8c6403e42db138f73e01b4c43b9235e62bfd36734727836895`.
7
+ - Exported tensor-file SHA256: `a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Input width: 2305. Gate: `base_minus_refined`, strict threshold `-0.06998290770476866`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
12
+
13
+ Project license is selected by the authors before public upload.
granular-relational/config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "granular-relational",
4
+ "source_sha256": "8f613280f8084e8c6403e42db138f73e01b4c43b9235e62bfd36734727836895",
5
+ "weights_sha256": "a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504",
6
+ "tensor_count": 30,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "granular",
9
+ "model": {
10
+ "input_dim": 2305,
11
+ "hidden_dim": 64,
12
+ "low_rank": 8,
13
+ "max_correction": 0.2
14
+ },
15
+ "input_dim": 2305,
16
+ "gate_threshold": -0.06998290770476866,
17
+ "gate_mode": "base_minus_refined",
18
+ "gate_decimals": null,
19
+ "fusion_weights": null,
20
+ "upstream_included": false,
21
+ "lineage": {
22
+ "schema": "dajepa_granular_train_lineage_v1",
23
+ "method": "da_jepa_r",
24
+ "dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
25
+ "config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
26
+ "adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
27
+ "train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
28
+ "calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
29
+ "train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
30
+ "calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
31
+ "trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
32
+ "core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
33
+ },
34
+ "training": {
35
+ "seed": 20260831,
36
+ "updates": 1500,
37
+ "batch_starts": 16,
38
+ "learning_rate": 0.0003,
39
+ "weight_decay": 0.0001,
40
+ "max_grad_norm": 1.0,
41
+ "tail_k": 16,
42
+ "evaluate_every": 25,
43
+ "mg_grid_step": 0.05,
44
+ "supervision_count": 16
45
+ }
46
+ }
granular-relational/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504
3
+ size 886593
manifest.json ADDED
@@ -0,0 +1,702 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "profiles": [
3
+ {
4
+ "project": "D-JEPA",
5
+ "profile": "granular-multiview",
6
+ "source_sha256": "e796fa3ee2fbc6f5b4abdfae99e8fa843606f923284a7fb95f59c0fc6c512c70",
7
+ "weights_sha256": "2ee5783def21816054645f87d9cffa50900733b7218a2b9ee203ae39aec682ff",
8
+ "tensor_count": 30,
9
+ "public_status": "local_release_candidate_license_not_yet_selected",
10
+ "architecture": "granular",
11
+ "model": {
12
+ "input_dim": 4,
13
+ "hidden_dim": 64,
14
+ "low_rank": 8,
15
+ "max_correction": 0.2
16
+ },
17
+ "input_dim": 4,
18
+ "gate_threshold": -0.010684727949480862,
19
+ "gate_mode": "base_minus_refined",
20
+ "gate_decimals": null,
21
+ "fusion_weights": [
22
+ 0.0,
23
+ 0.35,
24
+ 0.6,
25
+ 0.05
26
+ ],
27
+ "upstream_included": false,
28
+ "lineage": {
29
+ "schema": "dajepa_granular_train_lineage_v1",
30
+ "method": "da_jepa_mg",
31
+ "dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
32
+ "config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
33
+ "adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
34
+ "train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
35
+ "calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
36
+ "train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
37
+ "calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
38
+ "trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
39
+ "core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
40
+ },
41
+ "training": {
42
+ "seed": 20260831,
43
+ "updates": 1500,
44
+ "batch_starts": 16,
45
+ "learning_rate": 0.0003,
46
+ "weight_decay": 0.0001,
47
+ "max_grad_norm": 1.0,
48
+ "tail_k": 16,
49
+ "evaluate_every": 25,
50
+ "mg_grid_step": 0.05,
51
+ "supervision_count": 16
52
+ }
53
+ },
54
+ {
55
+ "project": "D-JEPA",
56
+ "profile": "granular-relational",
57
+ "source_sha256": "8f613280f8084e8c6403e42db138f73e01b4c43b9235e62bfd36734727836895",
58
+ "weights_sha256": "a7f2b70c1d1d853cf5dbd391f3560828f28015a72092938a978dff1a0e497504",
59
+ "tensor_count": 30,
60
+ "public_status": "local_release_candidate_license_not_yet_selected",
61
+ "architecture": "granular",
62
+ "model": {
63
+ "input_dim": 2305,
64
+ "hidden_dim": 64,
65
+ "low_rank": 8,
66
+ "max_correction": 0.2
67
+ },
68
+ "input_dim": 2305,
69
+ "gate_threshold": -0.06998290770476866,
70
+ "gate_mode": "base_minus_refined",
71
+ "gate_decimals": null,
72
+ "fusion_weights": null,
73
+ "upstream_included": false,
74
+ "lineage": {
75
+ "schema": "dajepa_granular_train_lineage_v1",
76
+ "method": "da_jepa_r",
77
+ "dino_checkpoint_sha256": "2b1cb1db727e866c9f98136a72dcd1e55ad076379bda64b4b88273e84a23ac60",
78
+ "config_sha256": "4018434a9031d3367c4067b3791f8afab7a31debaf7f264945146138f181a94e",
79
+ "adapter_config_sha256": "0f2c0a07bf170e87776bb5c5a3485a442ecfefa61eb0b06d3e143de218b000db",
80
+ "train_feature_cache_sha256": "45ec06b6a5331e26f454c3e33247308fe8da950b825e8ee1d349d7d2914ad40e",
81
+ "calibration_feature_cache_sha256": "be8936712c9bd828921f6f714b626928dab23760d2dcf35fb7419609d294ce65",
82
+ "train_release_sha256": "ac21f702f23b7a01f88aeb063785d24486face6ffd59b20b07de30cea81a4aa0",
83
+ "calibration_release_sha256": "67cb52466c3cfd3b1c35dcc89993960a925ab114d33c4c352b179763907828f7",
84
+ "trainer_sha256": "9686c9d555b41ad3966811452c514a3b26af88402b1b2d0ffc45be3d8528333d",
85
+ "core_sha256": "d267a9cdd5e2d8947f7bcb0d7bf49a0180d643a3a345e6bad3bc41b40df7ec81"
86
+ },
87
+ "training": {
88
+ "seed": 20260831,
89
+ "updates": 1500,
90
+ "batch_starts": 16,
91
+ "learning_rate": 0.0003,
92
+ "weight_decay": 0.0001,
93
+ "max_grad_norm": 1.0,
94
+ "tail_k": 16,
95
+ "evaluate_every": 25,
96
+ "mg_grid_step": 0.05,
97
+ "supervision_count": 16
98
+ }
99
+ },
100
+ {
101
+ "project": "D-JEPA",
102
+ "profile": "pushobj-unseen-shapes",
103
+ "source_sha256": "941dbd680a7dd179a798e208785bb2a68e3011753eb6d938f5f44a1f7c5ff404",
104
+ "weights_sha256": "8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024",
105
+ "tensor_count": 30,
106
+ "public_status": "local_release_candidate_license_not_yet_selected",
107
+ "architecture": "set_aligner",
108
+ "input_dim": 3,
109
+ "max_correction": 0.2,
110
+ "fusion_alpha": 0.3,
111
+ "gate_threshold": 0.0,
112
+ "gate_mode": "refined_gap",
113
+ "gate_decimals": null,
114
+ "upstream_included": false,
115
+ "training": {
116
+ "schema": "djepa-pushobj-relational-v1",
117
+ "seed": 20260906,
118
+ "epochs": 120,
119
+ "learning_rate": 0.0003,
120
+ "batch_size": 16,
121
+ "input_dim": 3,
122
+ "max_correction": 0.2,
123
+ "source_model": "frozen_pushobj_release",
124
+ "feature_mode": "ordinal",
125
+ "fusion_weight_selected_on_calibration": false,
126
+ "base_mode": "rank_fusion",
127
+ "fusion_alpha": 0.3,
128
+ "train_completion_sha256": "855bd0c8afe39add66b23fa5b6b5b401025ca1a6ef0d314dce253b1d5694e8c3",
129
+ "calibration_completion_sha256": "8ca0e50c69c47f0aed75e4c07b22f1ee00f0597621fc167b9da8110801b8c4e1",
130
+ "torch": "2.3.0+cu121",
131
+ "train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
132
+ "model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
133
+ },
134
+ "selected_epoch": 40
135
+ },
136
+ {
137
+ "project": "D-JEPA",
138
+ "profile": "pusht-adapted-predictor-full",
139
+ "source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
140
+ "weights_sha256": "26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083",
141
+ "tensor_count": 309,
142
+ "public_status": "local_release_candidate_license_not_yet_selected",
143
+ "architecture": "pretrained_predictor",
144
+ "upstream_included": true,
145
+ "lineage": {
146
+ "method": "dtail_jepa_p_v2",
147
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
148
+ "tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
149
+ "refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
150
+ "refiner_lineage": {
151
+ "method": "dtail_jepa_r_v2",
152
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
153
+ "lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
154
+ "tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
155
+ "fusion_alpha": 0.42,
156
+ "fit_start_count": 448,
157
+ "calibration_start_count": 64,
158
+ "best_update": 550
159
+ },
160
+ "trainable_names": [
161
+ "predictor.transformer.layers.5.attn.norm.weight",
162
+ "predictor.transformer.layers.5.attn.norm.bias",
163
+ "predictor.transformer.layers.5.attn.to_qkv.weight",
164
+ "predictor.transformer.layers.5.attn.to_out.0.weight",
165
+ "predictor.transformer.layers.5.attn.to_out.0.bias",
166
+ "predictor.transformer.layers.5.mlp.net.0.weight",
167
+ "predictor.transformer.layers.5.mlp.net.0.bias",
168
+ "predictor.transformer.layers.5.mlp.net.1.weight",
169
+ "predictor.transformer.layers.5.mlp.net.1.bias",
170
+ "predictor.transformer.layers.5.mlp.net.4.weight",
171
+ "predictor.transformer.layers.5.mlp.net.4.bias",
172
+ "predictor.transformer.layers.5.adaLN_modulation.1.weight",
173
+ "predictor.transformer.layers.5.adaLN_modulation.1.bias",
174
+ "pred_proj.net.0.weight",
175
+ "pred_proj.net.0.bias",
176
+ "pred_proj.net.1.weight",
177
+ "pred_proj.net.1.bias",
178
+ "pred_proj.net.3.weight",
179
+ "pred_proj.net.3.bias"
180
+ ],
181
+ "best_update": 200,
182
+ "fit_start_count": 384,
183
+ "calibration_start_count": 128,
184
+ "fusion_alpha": 0.42,
185
+ "config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
186
+ "code_sha256": {
187
+ "dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
188
+ "dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
189
+ "dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
190
+ "dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
191
+ "dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
192
+ "dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
193
+ "dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
194
+ "dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
195
+ "dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
196
+ "dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
197
+ "closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
198
+ "closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
199
+ "closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
200
+ },
201
+ "runtime_action_inputs": {
202
+ "artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
203
+ "sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
204
+ "validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
205
+ "completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
206
+ }
207
+ }
208
+ },
209
+ {
210
+ "project": "D-JEPA",
211
+ "profile": "pusht-exact-realization-full",
212
+ "source_sha256": "edf28ce23a2c9cc511c63f1f8d64f93e9c156436ff1fccfc1cecbc5545d55be2",
213
+ "weights_sha256": "983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2",
214
+ "tensor_count": 642,
215
+ "public_status": "local_release_candidate_license_not_yet_selected",
216
+ "architecture": "exact_realization_world_model",
217
+ "upstream_included": true,
218
+ "decision": {
219
+ "fusion_alpha": 0.42,
220
+ "gate_threshold": -0.03162526342176623,
221
+ "ordinal_denominator": 62,
222
+ "gate_comparison": "strict_gt",
223
+ "advantage_quantization": "decimal_half_even_4"
224
+ },
225
+ "lifting": {
226
+ "mode": "td_radial_ordinal_v1",
227
+ "radius_denominator": 64,
228
+ "zero_rms_threshold": 1e-12,
229
+ "preserved_prefix_steps": 4,
230
+ "native_cost_dtype": "float32"
231
+ },
232
+ "preprocessing": {
233
+ "image_mean": [
234
+ 0.485,
235
+ 0.456,
236
+ 0.406
237
+ ],
238
+ "image_std": [
239
+ 0.229,
240
+ 0.224,
241
+ 0.225
242
+ ],
243
+ "context_frames": 3,
244
+ "action_block": 5,
245
+ "future_blocks": 5,
246
+ "action_mean": [
247
+ -0.007812565192580223,
248
+ 0.006860686466097832
249
+ ],
250
+ "action_std": [
251
+ 0.2084674835205078,
252
+ 0.20674866437911987
253
+ ]
254
+ },
255
+ "lineage": {
256
+ "source_sha256": {
257
+ "tdjepa_weights": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
258
+ "tdjepa_config": "c7c802b3c96d5233ead1839c10fc3a85c0c484ce10a755af119a9ee5b1575075",
259
+ "lewm_object": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
260
+ "lewm_weights": "48938400ae3464c9680731287f583a9cb516f55a8ec64ea13a91be47fb15b607",
261
+ "lewm_config": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
262
+ "refiner": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
263
+ "refiner_config": "b1ebb7780bdb2ccfeb345b7f9ca5a3877becb760636a61d03daf2fb4d9387695",
264
+ "runtime_actions": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
265
+ "hard_suite": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba"
266
+ },
267
+ "packaging_config_sha256": "6932eedf77a4cd8d1f5cdf6e85961291ed10637bff7375c3800c8927e097e9ce",
268
+ "code_sha256": {
269
+ "dtail_wm/unified_r_model.py": "6efad8811121cbd5c218daad6e319dd967de1d989fb6bc23fbe4bb2f78f3f1c3",
270
+ "dtail_wm/package_unified_r.py": "13b0bd40ffe2851cddf566d87d2aadc833fed223b63974a6a751fe88f2276f58",
271
+ "reviewed_r/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
272
+ "reviewed_r/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
273
+ "reviewed_r/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
274
+ "tdjepa/jepa.py": "7de87eabd061b9af3186e57198ff951e747168e4b5260bd94a51db9bd18ec62d",
275
+ "tdjepa/module.py": "f7688fea10208c5127093bd1a7525935e12d2a6df0a0ae1f069f7c4dda99252a",
276
+ "lewm/lewm.py": "c93fe07c57fedfa183d0a68a57d4092c2fcd53a109901027a1798d5dbae4bd6c",
277
+ "lewm/module.py": "3a6209baf30c11f77a5f6637282a1821fc62cea827438a97b23933caab51d8c7",
278
+ "stable_pretraining/backbone/utils.py": "269b18ca0a8e0cc9e1c7909bb349b4b094243acdc80a3fb07e847d63d2c3303b"
279
+ },
280
+ "runtime_source_sha256": {
281
+ "snapshot_manifest": "8bc16050969b571d65c190e147ab59e63c1bce2b69b38ff2d179b271bff6b4f2"
282
+ },
283
+ "source_candidate_contract": "63_unique_positive_literal_ids_expert_zero_excluded",
284
+ "packaging_commit": "42676e0a98249ca62ae4646a1cc5826181c36a66",
285
+ "state_counts": {
286
+ "tdjepa": 309,
287
+ "lewm": 303,
288
+ "refiner": 30
289
+ },
290
+ "lewm_state_attestation": {
291
+ "sorted_key_sha256": "ade2562bd5d2be228bb9f8fb2fea26fedef83d2f8a84a4c74f469acfc18fd89d",
292
+ "tensor_content_sha256": "f6c74a65523cb0448d620a42fa76203f22b4e19d68661518aff93eecf2e3cb77"
293
+ }
294
+ },
295
+ "model_architecture": {
296
+ "candidate_count": 63,
297
+ "horizon": 5,
298
+ "latent_dim": 192,
299
+ "tdjepa_config": {
300
+ "_target_": "jepa.JEPA",
301
+ "encoder": {
302
+ "_target_": "stable_pretraining.backbone.utils.vit_hf",
303
+ "size": "tiny",
304
+ "patch_size": 14,
305
+ "image_size": 224,
306
+ "pretrained": false,
307
+ "use_mask_token": false
308
+ },
309
+ "predictor": {
310
+ "_target_": "module.ARPredictor",
311
+ "num_frames": 3,
312
+ "input_dim": 192,
313
+ "hidden_dim": 192,
314
+ "output_dim": 192,
315
+ "depth": 6,
316
+ "heads": 16,
317
+ "mlp_dim": 2048,
318
+ "dim_head": 64,
319
+ "dropout": 0.1,
320
+ "emb_dropout": 0.0
321
+ },
322
+ "action_encoder": {
323
+ "_target_": "module.Embedder",
324
+ "input_dim": 10,
325
+ "emb_dim": 192
326
+ },
327
+ "projector": {
328
+ "_target_": "module.MLP",
329
+ "input_dim": 192,
330
+ "output_dim": 192,
331
+ "hidden_dim": 2048,
332
+ "norm_fn": {
333
+ "_target_": "torch.nn.BatchNorm1d",
334
+ "_partial_": true
335
+ }
336
+ },
337
+ "pred_proj": {
338
+ "_target_": "module.MLP",
339
+ "input_dim": 192,
340
+ "output_dim": 192,
341
+ "hidden_dim": 2048,
342
+ "norm_fn": {
343
+ "_target_": "torch.nn.BatchNorm1d",
344
+ "_partial_": true
345
+ }
346
+ },
347
+ "planning_cost_mode": "td_jepa",
348
+ "dist_head": {
349
+ "_target_": "module.QuasimetricHead",
350
+ "input_dim": 192,
351
+ "hidden_dim": 512,
352
+ "sym_dim": 128,
353
+ "asym_dim": 128
354
+ },
355
+ "planning_cost_mse_blend": 0.1
356
+ },
357
+ "lewm_config": {
358
+ "_target_": "stable_worldmodel.wm.lewm.LeWM",
359
+ "encoder": {
360
+ "_target_": "stable_pretraining.backbone.utils.vit_hf",
361
+ "size": "tiny",
362
+ "patch_size": 14,
363
+ "image_size": 224,
364
+ "pretrained": false,
365
+ "use_mask_token": false
366
+ },
367
+ "predictor": {
368
+ "_target_": "stable_worldmodel.wm.lewm.module.Predictor",
369
+ "num_frames": 3,
370
+ "input_dim": 192,
371
+ "hidden_dim": 192,
372
+ "output_dim": 192,
373
+ "depth": 6,
374
+ "heads": 16,
375
+ "mlp_dim": 2048,
376
+ "dim_head": 64,
377
+ "dropout": 0.1,
378
+ "emb_dropout": 0.0
379
+ },
380
+ "action_encoder": {
381
+ "_target_": "stable_worldmodel.wm.lewm.module.Embedder",
382
+ "input_dim": 10,
383
+ "emb_dim": 192
384
+ },
385
+ "projector": {
386
+ "_target_": "stable_worldmodel.wm.lewm.module.MLP",
387
+ "input_dim": 192,
388
+ "output_dim": 192,
389
+ "hidden_dim": 2048,
390
+ "norm_fn": {
391
+ "_target_": "torch.nn.BatchNorm1d",
392
+ "_partial_": true
393
+ }
394
+ },
395
+ "pred_proj": {
396
+ "_target_": "stable_worldmodel.wm.lewm.module.MLP",
397
+ "input_dim": 192,
398
+ "output_dim": 192,
399
+ "hidden_dim": 2048,
400
+ "norm_fn": {
401
+ "_target_": "torch.nn.BatchNorm1d",
402
+ "_partial_": true
403
+ }
404
+ }
405
+ },
406
+ "refiner_config": {
407
+ "hidden_dim": 64,
408
+ "low_rank": 8,
409
+ "max_correction": 0.2
410
+ }
411
+ },
412
+ "deployment": "One file containing two pretrained predictors and the relational operator; not single-backbone distillation."
413
+ },
414
+ {
415
+ "project": "D-JEPA",
416
+ "profile": "pusht-multi-geometry",
417
+ "source_sha256": "d187c0efc194e6faa39007be835f3cca820b16fd95adfb1d7283f3d44d108b56",
418
+ "weights_sha256": "801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3",
419
+ "tensor_count": 30,
420
+ "public_status": "local_release_candidate_license_not_yet_selected",
421
+ "architecture": "set_aligner",
422
+ "input_dim": 388,
423
+ "max_correction": 0.2,
424
+ "gate_threshold": 0.027186155319213864,
425
+ "gate_mode": "base_minus_refined",
426
+ "gate_decimals": null,
427
+ "base_geometry": "jepa_wm",
428
+ "geometry_order": [
429
+ "lewm",
430
+ "temporal_distance",
431
+ "jepa_wm",
432
+ "dino_wm"
433
+ ],
434
+ "upstream_included": false,
435
+ "lineage": {
436
+ "schema": "dtail_multi_geometry_lineage_v1",
437
+ "scope": "locked_development",
438
+ "config_sha256": "16a4170801d3b84cfaea19565f1d7fb80663adcb30fa6164eeec3d54da7cba1d",
439
+ "runner_sha256": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
440
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
441
+ "registry_sha256": "84b7b1044b958a0bfc85f9838b8ff82568fad5c76ac742ac29a79cedf6b87f65",
442
+ "dependency_sha256": {
443
+ "source-file:multi_geometry.py": "e3a50a62fd36b4552ae19623691b044442af36bbb1c3925d410c027884d76ac2",
444
+ "source-file:physical_join.py": "c742a241edab03e43107f52a0207fa669a66e0fe430ec29736789892ca510c03",
445
+ "source-file:aggregate.py": "95b478b8f5a7827e3c1963176245d8e229fbca3f73798b71becbce02cb6c75ad",
446
+ "source-file:registry.py": "e7ccd4e18f708c31267f5681f56812cc811a489a84d9c2499cd0dca9d3d7d08f",
447
+ "source-file:releases.py": "3ecdb263a4e87bf3e5ba317d7baf15f648dd132d97cf641cc49e728678f632f2",
448
+ "source-file:score_io.py": "126a7b713a8d02c8b0af69922a1073495e45a1713b2ab44f2ca683a15e438beb",
449
+ "source-file:route_experiment.py": "ad558c3b16209ff6b8b709a645ab134e63fc8883397f8124552e80d4f0239090",
450
+ "source-file:run_multi_geometry_routes.py": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
451
+ "source-file:dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
452
+ "source-file:dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
453
+ "source-file:run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8"
454
+ },
455
+ "train_release_sha256": "3c6770e13abfaf4e890d0e273cf530e0b54df690f46683f7395c4c9a3b5e74bc",
456
+ "development_release_sha256": "5b0dbee6f1693692037ef65016678adeee7cfa7b08fb5646ff86dbc95a9e0c72",
457
+ "external_score_sha256": {
458
+ "train": {
459
+ "jepa_wm": "0c6d7efa4abb1c822d9511cf0ef36aed0f7e5acb157c0973e344313e8c2681e2",
460
+ "dino_wm": "5eaff47859b272b90c424893e41f281a66902cbc45e2b7162a06bc87d5e462a7"
461
+ },
462
+ "development": {
463
+ "jepa_wm": "7fc522ca6ddde02a4eb673a4e469b5a30e7a3510b5452d3bcdfedf0d3bd51c83",
464
+ "dino_wm": "80c4fb904fa12c305294431047962ae35e8bbf4810056d1085d1d80903491588"
465
+ }
466
+ },
467
+ "geometry_order": [
468
+ "lewm",
469
+ "temporal_distance",
470
+ "jepa_wm",
471
+ "dino_wm"
472
+ ],
473
+ "convex_weights": [
474
+ 0.0,
475
+ 0.0,
476
+ 0.25,
477
+ 0.75
478
+ ],
479
+ "fusion_train_correct": 444,
480
+ "fit_count": 448,
481
+ "method": "dtail_jepa_base_multi_geometry",
482
+ "initial_state_sha256": "947de445ddfe14280fa74a200bf12c56d47113a191942d1e5144a06dbff8a836",
483
+ "best_update": 900,
484
+ "threshold": 0.027186155319213864,
485
+ "calibration_correct": 64,
486
+ "calibration_count": 64
487
+ },
488
+ "evaluation_scope": "development_128"
489
+ },
490
+ {
491
+ "project": "D-JEPA",
492
+ "profile": "pusht-predictive-adaptation",
493
+ "source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
494
+ "weights_sha256": "b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a",
495
+ "tensor_count": 22,
496
+ "public_status": "local_release_candidate_license_not_yet_selected",
497
+ "architecture": "predictor_patch",
498
+ "upstream_included": false,
499
+ "base_checkpoint_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
500
+ "lineage": {
501
+ "method": "dtail_jepa_p_v2",
502
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
503
+ "tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
504
+ "refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
505
+ "refiner_lineage": {
506
+ "method": "dtail_jepa_r_v2",
507
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
508
+ "lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
509
+ "tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
510
+ "fusion_alpha": 0.42,
511
+ "fit_start_count": 448,
512
+ "calibration_start_count": 64,
513
+ "best_update": 550
514
+ },
515
+ "trainable_names": [
516
+ "predictor.transformer.layers.5.attn.norm.weight",
517
+ "predictor.transformer.layers.5.attn.norm.bias",
518
+ "predictor.transformer.layers.5.attn.to_qkv.weight",
519
+ "predictor.transformer.layers.5.attn.to_out.0.weight",
520
+ "predictor.transformer.layers.5.attn.to_out.0.bias",
521
+ "predictor.transformer.layers.5.mlp.net.0.weight",
522
+ "predictor.transformer.layers.5.mlp.net.0.bias",
523
+ "predictor.transformer.layers.5.mlp.net.1.weight",
524
+ "predictor.transformer.layers.5.mlp.net.1.bias",
525
+ "predictor.transformer.layers.5.mlp.net.4.weight",
526
+ "predictor.transformer.layers.5.mlp.net.4.bias",
527
+ "predictor.transformer.layers.5.adaLN_modulation.1.weight",
528
+ "predictor.transformer.layers.5.adaLN_modulation.1.bias",
529
+ "pred_proj.net.0.weight",
530
+ "pred_proj.net.0.bias",
531
+ "pred_proj.net.1.weight",
532
+ "pred_proj.net.1.bias",
533
+ "pred_proj.net.3.weight",
534
+ "pred_proj.net.3.bias"
535
+ ],
536
+ "best_update": 200,
537
+ "fit_start_count": 384,
538
+ "calibration_start_count": 128,
539
+ "fusion_alpha": 0.42,
540
+ "config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
541
+ "code_sha256": {
542
+ "dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
543
+ "dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
544
+ "dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
545
+ "dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
546
+ "dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
547
+ "dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
548
+ "dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
549
+ "dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
550
+ "dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
551
+ "dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
552
+ "closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
553
+ "closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
554
+ "closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
555
+ },
556
+ "runtime_action_inputs": {
557
+ "artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
558
+ "sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
559
+ "validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
560
+ "completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
561
+ }
562
+ },
563
+ "usage": "Load exact pretrained Temporal-Distance-JEPA then replace these parameters. This is a parameter patch, not a full predictor."
564
+ },
565
+ {
566
+ "project": "D-JEPA",
567
+ "profile": "pusht-relational",
568
+ "source_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
569
+ "weights_sha256": "efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941",
570
+ "tensor_count": 30,
571
+ "public_status": "local_release_candidate_license_not_yet_selected",
572
+ "architecture": "set_aligner",
573
+ "input_dim": 386,
574
+ "max_correction": 0.2,
575
+ "fusion_alpha": 0.42,
576
+ "gate_threshold": -0.03162526342176623,
577
+ "gate_mode": "base_minus_refined",
578
+ "gate_decimals": null,
579
+ "upstream_included": false,
580
+ "lineage": {
581
+ "method": "dtail_jepa_r_v2",
582
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
583
+ "lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
584
+ "tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
585
+ "fusion_alpha": 0.42,
586
+ "fit_start_count": 448,
587
+ "calibration_start_count": 64,
588
+ "best_update": 550
589
+ }
590
+ },
591
+ {
592
+ "project": "D-JEPA",
593
+ "profile": "pusht-visual-shifts",
594
+ "source_sha256": "c9552d2ab5d477d5c07d71f64ac66a315f3bbcf287f866c40d64e857c67bac22",
595
+ "weights_sha256": "cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3",
596
+ "tensor_count": 30,
597
+ "public_status": "local_release_candidate_license_not_yet_selected",
598
+ "architecture": "set_aligner",
599
+ "input_dim": 3,
600
+ "max_correction": 0.2,
601
+ "fusion_alpha": 0.5,
602
+ "gate_threshold": 0.01,
603
+ "gate_mode": "refined_gap",
604
+ "gate_decimals": null,
605
+ "upstream_included": false,
606
+ "training": {
607
+ "schema": "djepa-pushobj-relational-v1",
608
+ "seed": 20260906,
609
+ "epochs": 120,
610
+ "learning_rate": 0.0003,
611
+ "batch_size": 16,
612
+ "input_dim": 3,
613
+ "max_correction": 0.2,
614
+ "source_model": "frozen_pusht_visual_shift_release",
615
+ "feature_mode": "ordinal",
616
+ "fusion_weight_selected_on_calibration": true,
617
+ "base_mode": "rank_fusion",
618
+ "fusion_alpha": 0.5,
619
+ "train_completion_sha256": "f13d2fa6a3ff152532c1acad7ec209e389737cbbee27a7ee9fce46bc39819bb8",
620
+ "calibration_completion_sha256": "a8e5952e7f3a6e2000d075f9198bffa99ddae1bc94b3b027953c0431d5bb81b6",
621
+ "torch": "2.3.0+cu121",
622
+ "train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
623
+ "model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
624
+ },
625
+ "selected_epoch": 20
626
+ },
627
+ {
628
+ "project": "D-JEPA",
629
+ "profile": "reacher-relational",
630
+ "source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
631
+ "weights_sha256": "5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3",
632
+ "tensor_count": 30,
633
+ "public_status": "local_release_candidate_license_not_yet_selected",
634
+ "architecture": "set_aligner",
635
+ "input_dim": 386,
636
+ "max_correction": 0.2,
637
+ "fusion_alpha": 0.4,
638
+ "gate_threshold": 0.12511829772303182,
639
+ "gate_mode": "base_minus_refined",
640
+ "gate_decimals": 8,
641
+ "upstream_included": false,
642
+ "bindings": {
643
+ "training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
644
+ "full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
645
+ "capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
646
+ "b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
647
+ "packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
648
+ }
649
+ },
650
+ {
651
+ "project": "D-JEPA",
652
+ "profile": "reacher-temporal-transport",
653
+ "source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
654
+ "weights_sha256": "cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f",
655
+ "tensor_count": 4,
656
+ "public_status": "local_release_candidate_license_not_yet_selected",
657
+ "architecture": "temporal_transport",
658
+ "hidden_dim": 32,
659
+ "radius": 0.1,
660
+ "fusion_alpha": 0.4,
661
+ "gate_threshold": 0.12511829772303182,
662
+ "gate_mode": "base_minus_refined",
663
+ "gate_decimals": 8,
664
+ "upstream_included": false,
665
+ "bindings": {
666
+ "training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
667
+ "full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
668
+ "capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
669
+ "b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
670
+ "packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
671
+ },
672
+ "physical_selector": "relational score; transported distance is a distinct diagnostic"
673
+ },
674
+ {
675
+ "project": "D-JEPA",
676
+ "profile": "reacher-world-model-full",
677
+ "source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
678
+ "weights_sha256": "f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a",
679
+ "tensor_count": 646,
680
+ "public_status": "local_release_candidate_license_not_yet_selected",
681
+ "architecture": "reacher_world_model",
682
+ "fusion_alpha": 0.4,
683
+ "gate_threshold": 0.12511829772303182,
684
+ "gate_mode": "base_minus_refined",
685
+ "gate_decimals": 8,
686
+ "upstream_included": true,
687
+ "bindings": {
688
+ "training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
689
+ "full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
690
+ "capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
691
+ "b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
692
+ "packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
693
+ },
694
+ "state_groups": [
695
+ "tdjepa",
696
+ "lewm",
697
+ "adapters"
698
+ ],
699
+ "physical_selector": "relational score"
700
+ }
701
+ ]
702
+ }
pushobj-unseen-shapes/README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: pushobj-unseen-shapes
2
+
3
+ - Architecture: `set_aligner`.
4
+ - Tensor count: 30.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `941dbd680a7dd179a798e208785bb2a68e3011753eb6d938f5f44a1f7c5ff404`.
7
+ - Exported tensor-file SHA256: `8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Input width: 3. Gate: `refined_gap`, strict threshold `0.0`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
12
+
13
+ Project license is selected by the authors before public upload.
pushobj-unseen-shapes/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "pushobj-unseen-shapes",
4
+ "source_sha256": "941dbd680a7dd179a798e208785bb2a68e3011753eb6d938f5f44a1f7c5ff404",
5
+ "weights_sha256": "8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024",
6
+ "tensor_count": 30,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "set_aligner",
9
+ "input_dim": 3,
10
+ "max_correction": 0.2,
11
+ "fusion_alpha": 0.3,
12
+ "gate_threshold": 0.0,
13
+ "gate_mode": "refined_gap",
14
+ "gate_decimals": null,
15
+ "upstream_included": false,
16
+ "training": {
17
+ "schema": "djepa-pushobj-relational-v1",
18
+ "seed": 20260906,
19
+ "epochs": 120,
20
+ "learning_rate": 0.0003,
21
+ "batch_size": 16,
22
+ "input_dim": 3,
23
+ "max_correction": 0.2,
24
+ "source_model": "frozen_pushobj_release",
25
+ "feature_mode": "ordinal",
26
+ "fusion_weight_selected_on_calibration": false,
27
+ "base_mode": "rank_fusion",
28
+ "fusion_alpha": 0.3,
29
+ "train_completion_sha256": "855bd0c8afe39add66b23fa5b6b5b401025ca1a6ef0d314dce253b1d5694e8c3",
30
+ "calibration_completion_sha256": "8ca0e50c69c47f0aed75e4c07b22f1ee00f0597621fc167b9da8110801b8c4e1",
31
+ "torch": "2.3.0+cu121",
32
+ "train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
33
+ "model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
34
+ },
35
+ "selected_epoch": 40
36
+ }
pushobj-unseen-shapes/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8797c805e77fa56017f2ba9d1699f5ffd2bdc415cb2882e01322290ff6e32024
3
+ size 279233
pusht-adapted-predictor-full/README.md ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: pusht-adapted-predictor-full
2
+
3
+ - Architecture: `pretrained_predictor`.
4
+ - Tensor count: 309.
5
+ - Includes pretrained predictor weights: true.
6
+ - Original checkpoint SHA256: `ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887`.
7
+ - Exported tensor-file SHA256: `26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Publication hold: upstream weight redistribution terms must be confirmed before upload.
pusht-adapted-predictor-full/config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "pusht-adapted-predictor-full",
4
+ "source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
5
+ "weights_sha256": "26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083",
6
+ "tensor_count": 309,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "pretrained_predictor",
9
+ "upstream_included": true,
10
+ "lineage": {
11
+ "method": "dtail_jepa_p_v2",
12
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
13
+ "tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
14
+ "refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
15
+ "refiner_lineage": {
16
+ "method": "dtail_jepa_r_v2",
17
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
18
+ "lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
19
+ "tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
20
+ "fusion_alpha": 0.42,
21
+ "fit_start_count": 448,
22
+ "calibration_start_count": 64,
23
+ "best_update": 550
24
+ },
25
+ "trainable_names": [
26
+ "predictor.transformer.layers.5.attn.norm.weight",
27
+ "predictor.transformer.layers.5.attn.norm.bias",
28
+ "predictor.transformer.layers.5.attn.to_qkv.weight",
29
+ "predictor.transformer.layers.5.attn.to_out.0.weight",
30
+ "predictor.transformer.layers.5.attn.to_out.0.bias",
31
+ "predictor.transformer.layers.5.mlp.net.0.weight",
32
+ "predictor.transformer.layers.5.mlp.net.0.bias",
33
+ "predictor.transformer.layers.5.mlp.net.1.weight",
34
+ "predictor.transformer.layers.5.mlp.net.1.bias",
35
+ "predictor.transformer.layers.5.mlp.net.4.weight",
36
+ "predictor.transformer.layers.5.mlp.net.4.bias",
37
+ "predictor.transformer.layers.5.adaLN_modulation.1.weight",
38
+ "predictor.transformer.layers.5.adaLN_modulation.1.bias",
39
+ "pred_proj.net.0.weight",
40
+ "pred_proj.net.0.bias",
41
+ "pred_proj.net.1.weight",
42
+ "pred_proj.net.1.bias",
43
+ "pred_proj.net.3.weight",
44
+ "pred_proj.net.3.bias"
45
+ ],
46
+ "best_update": 200,
47
+ "fit_start_count": 384,
48
+ "calibration_start_count": 128,
49
+ "fusion_alpha": 0.42,
50
+ "config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
51
+ "code_sha256": {
52
+ "dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
53
+ "dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
54
+ "dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
55
+ "dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
56
+ "dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
57
+ "dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
58
+ "dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
59
+ "dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
60
+ "dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
61
+ "dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
62
+ "closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
63
+ "closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
64
+ "closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
65
+ },
66
+ "runtime_action_inputs": {
67
+ "artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
68
+ "sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
69
+ "validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
70
+ "completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
71
+ }
72
+ }
73
+ }
pusht-adapted-predictor-full/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26cb528b6ba5b3b6e7d464eeb9a75f599ac28641617666e0638c9e51de2b9083
3
+ size 73174815
pusht-exact-realization-full/README.md ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: pusht-exact-realization-full
2
+
3
+ - Architecture: `exact_realization_world_model`.
4
+ - Tensor count: 642.
5
+ - Includes pretrained predictor weights: true.
6
+ - Original checkpoint SHA256: `edf28ce23a2c9cc511c63f1f8d64f93e9c156436ff1fccfc1cecbc5545d55be2`.
7
+ - Exported tensor-file SHA256: `983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Publication hold: upstream weight redistribution terms must be confirmed before upload.
pusht-exact-realization-full/config.json ADDED
@@ -0,0 +1,205 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "pusht-exact-realization-full",
4
+ "source_sha256": "edf28ce23a2c9cc511c63f1f8d64f93e9c156436ff1fccfc1cecbc5545d55be2",
5
+ "weights_sha256": "983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2",
6
+ "tensor_count": 642,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "exact_realization_world_model",
9
+ "upstream_included": true,
10
+ "decision": {
11
+ "fusion_alpha": 0.42,
12
+ "gate_threshold": -0.03162526342176623,
13
+ "ordinal_denominator": 62,
14
+ "gate_comparison": "strict_gt",
15
+ "advantage_quantization": "decimal_half_even_4"
16
+ },
17
+ "lifting": {
18
+ "mode": "td_radial_ordinal_v1",
19
+ "radius_denominator": 64,
20
+ "zero_rms_threshold": 1e-12,
21
+ "preserved_prefix_steps": 4,
22
+ "native_cost_dtype": "float32"
23
+ },
24
+ "preprocessing": {
25
+ "image_mean": [
26
+ 0.485,
27
+ 0.456,
28
+ 0.406
29
+ ],
30
+ "image_std": [
31
+ 0.229,
32
+ 0.224,
33
+ 0.225
34
+ ],
35
+ "context_frames": 3,
36
+ "action_block": 5,
37
+ "future_blocks": 5,
38
+ "action_mean": [
39
+ -0.007812565192580223,
40
+ 0.006860686466097832
41
+ ],
42
+ "action_std": [
43
+ 0.2084674835205078,
44
+ 0.20674866437911987
45
+ ]
46
+ },
47
+ "lineage": {
48
+ "source_sha256": {
49
+ "tdjepa_weights": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
50
+ "tdjepa_config": "c7c802b3c96d5233ead1839c10fc3a85c0c484ce10a755af119a9ee5b1575075",
51
+ "lewm_object": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
52
+ "lewm_weights": "48938400ae3464c9680731287f583a9cb516f55a8ec64ea13a91be47fb15b607",
53
+ "lewm_config": "2564086e961e7b5c7c04dffc451091115b389a590645ff19653c64fd0bc16e09",
54
+ "refiner": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
55
+ "refiner_config": "b1ebb7780bdb2ccfeb345b7f9ca5a3877becb760636a61d03daf2fb4d9387695",
56
+ "runtime_actions": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
57
+ "hard_suite": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba"
58
+ },
59
+ "packaging_config_sha256": "6932eedf77a4cd8d1f5cdf6e85961291ed10637bff7375c3800c8927e097e9ce",
60
+ "code_sha256": {
61
+ "dtail_wm/unified_r_model.py": "6efad8811121cbd5c218daad6e319dd967de1d989fb6bc23fbe4bb2f78f3f1c3",
62
+ "dtail_wm/package_unified_r.py": "13b0bd40ffe2851cddf566d87d2aadc833fed223b63974a6a751fe88f2276f58",
63
+ "reviewed_r/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
64
+ "reviewed_r/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
65
+ "reviewed_r/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
66
+ "tdjepa/jepa.py": "7de87eabd061b9af3186e57198ff951e747168e4b5260bd94a51db9bd18ec62d",
67
+ "tdjepa/module.py": "f7688fea10208c5127093bd1a7525935e12d2a6df0a0ae1f069f7c4dda99252a",
68
+ "lewm/lewm.py": "c93fe07c57fedfa183d0a68a57d4092c2fcd53a109901027a1798d5dbae4bd6c",
69
+ "lewm/module.py": "3a6209baf30c11f77a5f6637282a1821fc62cea827438a97b23933caab51d8c7",
70
+ "stable_pretraining/backbone/utils.py": "269b18ca0a8e0cc9e1c7909bb349b4b094243acdc80a3fb07e847d63d2c3303b"
71
+ },
72
+ "runtime_source_sha256": {
73
+ "snapshot_manifest": "8bc16050969b571d65c190e147ab59e63c1bce2b69b38ff2d179b271bff6b4f2"
74
+ },
75
+ "source_candidate_contract": "63_unique_positive_literal_ids_expert_zero_excluded",
76
+ "packaging_commit": "42676e0a98249ca62ae4646a1cc5826181c36a66",
77
+ "state_counts": {
78
+ "tdjepa": 309,
79
+ "lewm": 303,
80
+ "refiner": 30
81
+ },
82
+ "lewm_state_attestation": {
83
+ "sorted_key_sha256": "ade2562bd5d2be228bb9f8fb2fea26fedef83d2f8a84a4c74f469acfc18fd89d",
84
+ "tensor_content_sha256": "f6c74a65523cb0448d620a42fa76203f22b4e19d68661518aff93eecf2e3cb77"
85
+ }
86
+ },
87
+ "model_architecture": {
88
+ "candidate_count": 63,
89
+ "horizon": 5,
90
+ "latent_dim": 192,
91
+ "tdjepa_config": {
92
+ "_target_": "jepa.JEPA",
93
+ "encoder": {
94
+ "_target_": "stable_pretraining.backbone.utils.vit_hf",
95
+ "size": "tiny",
96
+ "patch_size": 14,
97
+ "image_size": 224,
98
+ "pretrained": false,
99
+ "use_mask_token": false
100
+ },
101
+ "predictor": {
102
+ "_target_": "module.ARPredictor",
103
+ "num_frames": 3,
104
+ "input_dim": 192,
105
+ "hidden_dim": 192,
106
+ "output_dim": 192,
107
+ "depth": 6,
108
+ "heads": 16,
109
+ "mlp_dim": 2048,
110
+ "dim_head": 64,
111
+ "dropout": 0.1,
112
+ "emb_dropout": 0.0
113
+ },
114
+ "action_encoder": {
115
+ "_target_": "module.Embedder",
116
+ "input_dim": 10,
117
+ "emb_dim": 192
118
+ },
119
+ "projector": {
120
+ "_target_": "module.MLP",
121
+ "input_dim": 192,
122
+ "output_dim": 192,
123
+ "hidden_dim": 2048,
124
+ "norm_fn": {
125
+ "_target_": "torch.nn.BatchNorm1d",
126
+ "_partial_": true
127
+ }
128
+ },
129
+ "pred_proj": {
130
+ "_target_": "module.MLP",
131
+ "input_dim": 192,
132
+ "output_dim": 192,
133
+ "hidden_dim": 2048,
134
+ "norm_fn": {
135
+ "_target_": "torch.nn.BatchNorm1d",
136
+ "_partial_": true
137
+ }
138
+ },
139
+ "planning_cost_mode": "td_jepa",
140
+ "dist_head": {
141
+ "_target_": "module.QuasimetricHead",
142
+ "input_dim": 192,
143
+ "hidden_dim": 512,
144
+ "sym_dim": 128,
145
+ "asym_dim": 128
146
+ },
147
+ "planning_cost_mse_blend": 0.1
148
+ },
149
+ "lewm_config": {
150
+ "_target_": "stable_worldmodel.wm.lewm.LeWM",
151
+ "encoder": {
152
+ "_target_": "stable_pretraining.backbone.utils.vit_hf",
153
+ "size": "tiny",
154
+ "patch_size": 14,
155
+ "image_size": 224,
156
+ "pretrained": false,
157
+ "use_mask_token": false
158
+ },
159
+ "predictor": {
160
+ "_target_": "stable_worldmodel.wm.lewm.module.Predictor",
161
+ "num_frames": 3,
162
+ "input_dim": 192,
163
+ "hidden_dim": 192,
164
+ "output_dim": 192,
165
+ "depth": 6,
166
+ "heads": 16,
167
+ "mlp_dim": 2048,
168
+ "dim_head": 64,
169
+ "dropout": 0.1,
170
+ "emb_dropout": 0.0
171
+ },
172
+ "action_encoder": {
173
+ "_target_": "stable_worldmodel.wm.lewm.module.Embedder",
174
+ "input_dim": 10,
175
+ "emb_dim": 192
176
+ },
177
+ "projector": {
178
+ "_target_": "stable_worldmodel.wm.lewm.module.MLP",
179
+ "input_dim": 192,
180
+ "output_dim": 192,
181
+ "hidden_dim": 2048,
182
+ "norm_fn": {
183
+ "_target_": "torch.nn.BatchNorm1d",
184
+ "_partial_": true
185
+ }
186
+ },
187
+ "pred_proj": {
188
+ "_target_": "stable_worldmodel.wm.lewm.module.MLP",
189
+ "input_dim": 192,
190
+ "output_dim": 192,
191
+ "hidden_dim": 2048,
192
+ "norm_fn": {
193
+ "_target_": "torch.nn.BatchNorm1d",
194
+ "_partial_": true
195
+ }
196
+ }
197
+ },
198
+ "refiner_config": {
199
+ "hidden_dim": 64,
200
+ "low_rank": 8,
201
+ "max_correction": 0.2
202
+ }
203
+ },
204
+ "deployment": "One file containing two pretrained predictors and the relational operator; not single-backbone distillation."
205
+ }
pusht-exact-realization-full/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:983be0921f3b1d393ce9c6b0e6acfda8272b546ec7009e93ec0a4466a91a1ba2
3
+ size 145803819
pusht-multi-geometry/README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: pusht-multi-geometry
2
+
3
+ - Architecture: `set_aligner`.
4
+ - Tensor count: 30.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `d187c0efc194e6faa39007be835f3cca820b16fd95adfb1d7283f3d44d108b56`.
7
+ - Exported tensor-file SHA256: `801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Input width: 388. Gate: `base_minus_refined`, strict threshold `0.027186155319213864`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
12
+
13
+ Project license is selected by the authors before public upload.
pusht-multi-geometry/config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "pusht-multi-geometry",
4
+ "source_sha256": "d187c0efc194e6faa39007be835f3cca820b16fd95adfb1d7283f3d44d108b56",
5
+ "weights_sha256": "801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3",
6
+ "tensor_count": 30,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "set_aligner",
9
+ "input_dim": 388,
10
+ "max_correction": 0.2,
11
+ "gate_threshold": 0.027186155319213864,
12
+ "gate_mode": "base_minus_refined",
13
+ "gate_decimals": null,
14
+ "base_geometry": "jepa_wm",
15
+ "geometry_order": [
16
+ "lewm",
17
+ "temporal_distance",
18
+ "jepa_wm",
19
+ "dino_wm"
20
+ ],
21
+ "upstream_included": false,
22
+ "lineage": {
23
+ "schema": "dtail_multi_geometry_lineage_v1",
24
+ "scope": "locked_development",
25
+ "config_sha256": "16a4170801d3b84cfaea19565f1d7fb80663adcb30fa6164eeec3d54da7cba1d",
26
+ "runner_sha256": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
27
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
28
+ "registry_sha256": "84b7b1044b958a0bfc85f9838b8ff82568fad5c76ac742ac29a79cedf6b87f65",
29
+ "dependency_sha256": {
30
+ "source-file:multi_geometry.py": "e3a50a62fd36b4552ae19623691b044442af36bbb1c3925d410c027884d76ac2",
31
+ "source-file:physical_join.py": "c742a241edab03e43107f52a0207fa669a66e0fe430ec29736789892ca510c03",
32
+ "source-file:aggregate.py": "95b478b8f5a7827e3c1963176245d8e229fbca3f73798b71becbce02cb6c75ad",
33
+ "source-file:registry.py": "e7ccd4e18f708c31267f5681f56812cc811a489a84d9c2499cd0dca9d3d7d08f",
34
+ "source-file:releases.py": "3ecdb263a4e87bf3e5ba317d7baf15f648dd132d97cf641cc49e728678f632f2",
35
+ "source-file:score_io.py": "126a7b713a8d02c8b0af69922a1073495e45a1713b2ab44f2ca683a15e438beb",
36
+ "source-file:route_experiment.py": "ad558c3b16209ff6b8b709a645ab134e63fc8883397f8124552e80d4f0239090",
37
+ "source-file:run_multi_geometry_routes.py": "7dff9d912ec14c1a8dc540907626b203843ff5cb632a9fcb730d1e89cf0ebfe9",
38
+ "source-file:dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
39
+ "source-file:dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
40
+ "source-file:run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8"
41
+ },
42
+ "train_release_sha256": "3c6770e13abfaf4e890d0e273cf530e0b54df690f46683f7395c4c9a3b5e74bc",
43
+ "development_release_sha256": "5b0dbee6f1693692037ef65016678adeee7cfa7b08fb5646ff86dbc95a9e0c72",
44
+ "external_score_sha256": {
45
+ "train": {
46
+ "jepa_wm": "0c6d7efa4abb1c822d9511cf0ef36aed0f7e5acb157c0973e344313e8c2681e2",
47
+ "dino_wm": "5eaff47859b272b90c424893e41f281a66902cbc45e2b7162a06bc87d5e462a7"
48
+ },
49
+ "development": {
50
+ "jepa_wm": "7fc522ca6ddde02a4eb673a4e469b5a30e7a3510b5452d3bcdfedf0d3bd51c83",
51
+ "dino_wm": "80c4fb904fa12c305294431047962ae35e8bbf4810056d1085d1d80903491588"
52
+ }
53
+ },
54
+ "geometry_order": [
55
+ "lewm",
56
+ "temporal_distance",
57
+ "jepa_wm",
58
+ "dino_wm"
59
+ ],
60
+ "convex_weights": [
61
+ 0.0,
62
+ 0.0,
63
+ 0.25,
64
+ 0.75
65
+ ],
66
+ "fusion_train_correct": 444,
67
+ "fit_count": 448,
68
+ "method": "dtail_jepa_base_multi_geometry",
69
+ "initial_state_sha256": "947de445ddfe14280fa74a200bf12c56d47113a191942d1e5144a06dbff8a836",
70
+ "best_update": 900,
71
+ "threshold": 0.027186155319213864,
72
+ "calibration_correct": 64,
73
+ "calibration_count": 64
74
+ },
75
+ "evaluation_scope": "development_128"
76
+ }
pusht-multi-geometry/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:801c92fc312b071fbc43b1e2153ae1ca25c08b1a3ae95110b38c7482e99c0fe3
3
+ size 377793
pusht-predictive-adaptation/README.md ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: pusht-predictive-adaptation
2
+
3
+ - Architecture: `predictor_patch`.
4
+ - Tensor count: 22.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887`.
7
+ - Exported tensor-file SHA256: `b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Project license is selected by the authors before public upload.
pusht-predictive-adaptation/config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "pusht-predictive-adaptation",
4
+ "source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
5
+ "weights_sha256": "b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a",
6
+ "tensor_count": 22,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "predictor_patch",
9
+ "upstream_included": false,
10
+ "base_checkpoint_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
11
+ "lineage": {
12
+ "method": "dtail_jepa_p_v2",
13
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
14
+ "tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
15
+ "refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
16
+ "refiner_lineage": {
17
+ "method": "dtail_jepa_r_v2",
18
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
19
+ "lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
20
+ "tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
21
+ "fusion_alpha": 0.42,
22
+ "fit_start_count": 448,
23
+ "calibration_start_count": 64,
24
+ "best_update": 550
25
+ },
26
+ "trainable_names": [
27
+ "predictor.transformer.layers.5.attn.norm.weight",
28
+ "predictor.transformer.layers.5.attn.norm.bias",
29
+ "predictor.transformer.layers.5.attn.to_qkv.weight",
30
+ "predictor.transformer.layers.5.attn.to_out.0.weight",
31
+ "predictor.transformer.layers.5.attn.to_out.0.bias",
32
+ "predictor.transformer.layers.5.mlp.net.0.weight",
33
+ "predictor.transformer.layers.5.mlp.net.0.bias",
34
+ "predictor.transformer.layers.5.mlp.net.1.weight",
35
+ "predictor.transformer.layers.5.mlp.net.1.bias",
36
+ "predictor.transformer.layers.5.mlp.net.4.weight",
37
+ "predictor.transformer.layers.5.mlp.net.4.bias",
38
+ "predictor.transformer.layers.5.adaLN_modulation.1.weight",
39
+ "predictor.transformer.layers.5.adaLN_modulation.1.bias",
40
+ "pred_proj.net.0.weight",
41
+ "pred_proj.net.0.bias",
42
+ "pred_proj.net.1.weight",
43
+ "pred_proj.net.1.bias",
44
+ "pred_proj.net.3.weight",
45
+ "pred_proj.net.3.bias"
46
+ ],
47
+ "best_update": 200,
48
+ "fit_start_count": 384,
49
+ "calibration_start_count": 128,
50
+ "fusion_alpha": 0.42,
51
+ "config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
52
+ "code_sha256": {
53
+ "dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
54
+ "dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
55
+ "dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
56
+ "dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
57
+ "dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
58
+ "dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
59
+ "dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
60
+ "dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
61
+ "dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
62
+ "dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
63
+ "closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
64
+ "closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
65
+ "closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
66
+ },
67
+ "runtime_action_inputs": {
68
+ "artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
69
+ "sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
70
+ "validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
71
+ "completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
72
+ }
73
+ },
74
+ "usage": "Load exact pretrained Temporal-Distance-JEPA then replace these parameters. This is a parameter patch, not a full predictor."
75
+ }
pusht-predictive-adaptation/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a
3
+ size 10387305
pusht-relational/README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: pusht-relational
2
+
3
+ - Architecture: `set_aligner`.
4
+ - Tensor count: 30.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f`.
7
+ - Exported tensor-file SHA256: `efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Input width: 386. Gate: `base_minus_refined`, strict threshold `-0.03162526342176623`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
12
+
13
+ Project license is selected by the authors before public upload.
pusht-relational/config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "pusht-relational",
4
+ "source_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
5
+ "weights_sha256": "efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941",
6
+ "tensor_count": 30,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "set_aligner",
9
+ "input_dim": 386,
10
+ "max_correction": 0.2,
11
+ "fusion_alpha": 0.42,
12
+ "gate_threshold": -0.03162526342176623,
13
+ "gate_mode": "base_minus_refined",
14
+ "gate_decimals": null,
15
+ "upstream_included": false,
16
+ "lineage": {
17
+ "method": "dtail_jepa_r_v2",
18
+ "pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
19
+ "lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
20
+ "tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
21
+ "fusion_alpha": 0.42,
22
+ "fit_start_count": 448,
23
+ "calibration_start_count": 64,
24
+ "best_update": 550
25
+ }
26
+ }
pusht-relational/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efea4301f605f681c7f85f934a6ecdcafa9984058f72be3be2d625bfc9e16941
3
+ size 377281
pusht-visual-shifts/README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: pusht-visual-shifts
2
+
3
+ - Architecture: `set_aligner`.
4
+ - Tensor count: 30.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `c9552d2ab5d477d5c07d71f64ac66a315f3bbcf287f866c40d64e857c67bac22`.
7
+ - Exported tensor-file SHA256: `cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Input width: 3. Gate: `refined_gap`, strict threshold `0.01`; quantization decimals: `None`. Candidate IDs and feature preprocessing must match the task profile.
12
+
13
+ Project license is selected by the authors before public upload.
pusht-visual-shifts/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "pusht-visual-shifts",
4
+ "source_sha256": "c9552d2ab5d477d5c07d71f64ac66a315f3bbcf287f866c40d64e857c67bac22",
5
+ "weights_sha256": "cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3",
6
+ "tensor_count": 30,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "set_aligner",
9
+ "input_dim": 3,
10
+ "max_correction": 0.2,
11
+ "fusion_alpha": 0.5,
12
+ "gate_threshold": 0.01,
13
+ "gate_mode": "refined_gap",
14
+ "gate_decimals": null,
15
+ "upstream_included": false,
16
+ "training": {
17
+ "schema": "djepa-pushobj-relational-v1",
18
+ "seed": 20260906,
19
+ "epochs": 120,
20
+ "learning_rate": 0.0003,
21
+ "batch_size": 16,
22
+ "input_dim": 3,
23
+ "max_correction": 0.2,
24
+ "source_model": "frozen_pusht_visual_shift_release",
25
+ "feature_mode": "ordinal",
26
+ "fusion_weight_selected_on_calibration": true,
27
+ "base_mode": "rank_fusion",
28
+ "fusion_alpha": 0.5,
29
+ "train_completion_sha256": "f13d2fa6a3ff152532c1acad7ec209e389737cbbee27a7ee9fce46bc39819bb8",
30
+ "calibration_completion_sha256": "a8e5952e7f3a6e2000d075f9198bffa99ddae1bc94b3b027953c0431d5bb81b6",
31
+ "torch": "2.3.0+cu121",
32
+ "train_code_sha256": "b890ce79e62d3787f405f14971dee1acdd1edc2e9f58b33ae9146b1ccc5e3247",
33
+ "model_code_sha256": "4db0e1740c89fdcbaea98cd07abd1f5d0fbdee85abb42d3791524dc6a6a8bc86"
34
+ },
35
+ "selected_epoch": 20
36
+ }
pusht-visual-shifts/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf36d6bd0af016d8adf2358d78feeda33c1bbf062eef1d0be6332ca1bb28f2e3
3
+ size 279233
reacher-relational/README.md ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: reacher-relational
2
+
3
+ - Architecture: `set_aligner`.
4
+ - Tensor count: 30.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05`.
7
+ - Exported tensor-file SHA256: `5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Input width: 386. Gate: `base_minus_refined`, strict threshold `0.12511829772303182`; quantization decimals: `8`. Candidate IDs and feature preprocessing must match the task profile.
12
+
13
+ Project license is selected by the authors before public upload.
reacher-relational/config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "reacher-relational",
4
+ "source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
5
+ "weights_sha256": "5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3",
6
+ "tensor_count": 30,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "set_aligner",
9
+ "input_dim": 386,
10
+ "max_correction": 0.2,
11
+ "fusion_alpha": 0.4,
12
+ "gate_threshold": 0.12511829772303182,
13
+ "gate_mode": "base_minus_refined",
14
+ "gate_decimals": 8,
15
+ "upstream_included": false,
16
+ "bindings": {
17
+ "training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
18
+ "full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
19
+ "capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
20
+ "b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
21
+ "packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
22
+ }
23
+ }
reacher-relational/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5160c04ac103589b33820133b0cb221189931382c3e6ef3bf21d9232bed830d3
3
+ size 377281
reacher-temporal-transport/README.md ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: reacher-temporal-transport
2
+
3
+ - Architecture: `temporal_transport`.
4
+ - Tensor count: 4.
5
+ - Includes pretrained predictor weights: false.
6
+ - Original checkpoint SHA256: `2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05`.
7
+ - Exported tensor-file SHA256: `cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Project license is selected by the authors before public upload.
reacher-temporal-transport/config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "reacher-temporal-transport",
4
+ "source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
5
+ "weights_sha256": "cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f",
6
+ "tensor_count": 4,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "temporal_transport",
9
+ "hidden_dim": 32,
10
+ "radius": 0.1,
11
+ "fusion_alpha": 0.4,
12
+ "gate_threshold": 0.12511829772303182,
13
+ "gate_mode": "base_minus_refined",
14
+ "gate_decimals": 8,
15
+ "upstream_included": false,
16
+ "bindings": {
17
+ "training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
18
+ "full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
19
+ "capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
20
+ "b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
21
+ "packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
22
+ },
23
+ "physical_selector": "relational score; transported distance is a distinct diagnostic"
24
+ }
reacher-temporal-transport/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf8c57a00e891fc002f9e595e3cf89c8a52a163e42c57059919bc1cfdddd883f
3
+ size 3081
reacher-world-model-full/README.md ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # D-JEPA: reacher-world-model-full
2
+
3
+ - Architecture: `reacher_world_model`.
4
+ - Tensor count: 646.
5
+ - Includes pretrained predictor weights: true.
6
+ - Original checkpoint SHA256: `2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05`.
7
+ - Exported tensor-file SHA256: `f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a`.
8
+
9
+ Load `model.pt` with `torch.load(..., map_location="cpu", weights_only=True)`. Use the matching D-JEPA code profile and JSON configuration. Tensor values are unchanged; metadata and optimizer state have been separated.
10
+
11
+ Publication hold: upstream weight redistribution terms must be confirmed before upload.
reacher-world-model-full/config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "project": "D-JEPA",
3
+ "profile": "reacher-world-model-full",
4
+ "source_sha256": "2bca723eaca2ef1dd7d01182b1f6f0b545aba646076b9e8ce7ba6f9f119bce05",
5
+ "weights_sha256": "f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a",
6
+ "tensor_count": 646,
7
+ "public_status": "local_release_candidate_license_not_yet_selected",
8
+ "architecture": "reacher_world_model",
9
+ "fusion_alpha": 0.4,
10
+ "gate_threshold": 0.12511829772303182,
11
+ "gate_mode": "base_minus_refined",
12
+ "gate_decimals": 8,
13
+ "upstream_included": true,
14
+ "bindings": {
15
+ "training_completion_sha256": "9af702ae3c9f3321863fc96c7f7dd1132748f6846cc07e289f8d28657fd7c636",
16
+ "full_pool_completion_sha256": "839dbbd1ace0dde4492b1d4e5a167fab12814f1330db0c00ea9d301639bdd964",
17
+ "capacity_gate_completion_sha256": "d6bab3c79667923f85efdb9e187f80eb20739db200ec83452efbee6edc47f6b2",
18
+ "b2_config_sha256": "737a766a978f16847ba8dbb240ce13f11bcda28b6d23b19f38082218563c6f73",
19
+ "packager_code_sha256": "4d8cbd215a0cf852a90d355be0dbeab661d6aeff17f4cc6eb6f01319446f8298"
20
+ },
21
+ "state_groups": [
22
+ "tdjepa",
23
+ "lewm",
24
+ "adapters"
25
+ ],
26
+ "physical_selector": "relational score"
27
+ }
reacher-world-model-full/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7331bf87bf4c8543fa8797072e660266e55d9a9354702120cec1faadd76609a
3
+ size 145806043