plastic: phys_mps_3k

Canonical repository: Both checkpoints now live together at dmontgomery40/plastic. Use the physics folder. This earlier release remains available for existing links.

Hidden-friction 2D point-mass system identification. Inputs are observation (4), action (2), reset flag (1); outputs are next-observation deltas (4). Hidden friction is not an input.

This is the implemented selective-recurrence plus linear delta-memory baseline with normalized readout, not the experimental nonlinear coordinate proposal or an exact reproduction of Sun et al. TTT-Linear. Outer training differentiates through the fast-memory updates. The external transactional harness is a separate runtime component.

Files and provenance

Checkpoint: 3,564,764 parameters, step 3000. Config, saved evaluation, license, and SHA-256 manifest are included; the text package also includes its BPE tokenizer. Training source revision is not recorded in this package; do not treat the compatible runtime revision as the training revision.

Compatible source: plastic at 269ecc9. Use Python 3.12+ and the dependencies pinned or declared by that checkout. This is custom PyTorch code, not a Transformers AutoModel package.

Load locally

Clone the source repository, check out the revision above, and run uv sync --extra dev. Run the example below with uv run python; it downloads this public checkpoint from Hugging Face:

from pathlib import Path
from huggingface_hub import snapshot_download
import torch
from plastic.config import ModelConfig
from plastic.model.lm import build_model

folder = Path(snapshot_download("dmontgomery40/plastic-physics-3.56m"))
checkpoint = torch.load(folder / "checkpoint.pt", map_location="cpu", weights_only=True)
cfg = ModelConfig.from_dict(checkpoint["config"])
model = build_model(cfg).eval()
model.load_state_dict(checkpoint["model_state"])
x = torch.zeros(1, 4, cfg.input_dim)
x[:, 0, -1] = 1  # new episode
with torch.no_grad():
    predictions, state, signals = model(x)
# Pass state to the next call to continue the same sequence.

This example directly uses the model and permits fast-memory updates. It does not apply the transactional safety harness. Calibration/canaries/Fisher files and session data are not included in this portable checkpoint package. Follow the source project's calibration workflow before creating guarded sessions.

Recorded evaluation

Metric: observation-delta MSE, over 131,072 target elements. Writes enabled: 0.0001288502863; writes disabled (beta_scale=0): 0.2087303363. These are saved single-run evaluations, not freshly reproduced by this packaging step. Disabling writes leaves retention active. Low prediction error does not by itself identify where friction is represented.

These observations support task-specific memory utility. They do not establish architectural novelty, general intelligence, or adversarial robustness. The CPU loading check establishes serialization/runtime compatibility only, not GPU/MPS performance. See the source repository for experimental methods and limitations.

License

The included license permits noncommercial use and requires prior written permission for commercial use. It is not the standard MIT license.

Downloads last month
23
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support