{
  "name": "Nexa_Mat Gen Stack",
  "model_id": "AethronPhantom/Nexa_Mat2",
  "published_at": "2026-05-20T01:11:29Z",
  "components": {
    "encoder": {
      "component": "nexa_mat_encoder_v1",
      "role": "frozen physics-grounding encoder",
      "repo_id": "AethronPhantom/Nexa_Mat2",
      "path": "encoder/v1/nexa_mat_V1_final.safetensors",
      "artifact_uri": "hf://AethronPhantom/Nexa_Mat2/encoder/v1/nexa_mat_V1_final.safetensors",
      "source_checkpoint_uri": "wasabi:nexa-mat/models/nexa_mat/encoder-pretrain/full_v1/2026-05-06/nexa_mat_encoder_blackwell_full_v1_continue_to_200k_20260506T075034Z/nexa_mat_V1_final.pt",
      "published_at": "2026-05-20T01:11:29Z",
      "format": "safetensors_model_state_only",
      "stats": {
        "bytes": 11264456,
        "sha256": "a39e1db3c86622cc36bc2610a47af0f6336578fc3b644a7068ed62190853e9a0"
      },
      "architecture": {
        "family": "directional_potential",
        "hidden_dim": 192,
        "egnn_layers": 4,
        "max_neighbors": 8,
        "radial_basis": 64,
        "cutoff_angstrom": 6.0,
        "message_aggregation": "learned",
        "energy_mode": "atomwise_sum",
        "absolute_coords": false,
        "parameters": 2813714
      },
      "training": {
        "phase": "encoder_full_pretrain_v1",
        "steps": 200000,
        "dataset": "nexa_mat_v1 force-compatible OMat24 surface",
        "science_status": "frozen_for_downstream_generative_experiments"
      }
    },
    "diffusion_decoder": {
      "component": "nexa_mat_diffusion_v1",
      "role": "constrained best-of-N diffusion proposal decoder",
      "repo_id": "AethronPhantom/Nexa_Mat2",
      "path": "decoder/diffusion_v1/final_checkpoint.safetensors",
      "artifact_uri": "hf://AethronPhantom/Nexa_Mat2/decoder/diffusion_v1/final_checkpoint.safetensors",
      "source_checkpoint_uri": "wasabi:nexa-mat/models/nexa_mat/diffusion-generation-v1/prod/nexa_mat_diffusion_prod_fp32_b512_phase_sampler_20260512T162654Z/final_checkpoint.pt",
      "published_at": "2026-05-20T01:11:29Z",
      "format": "safetensors_model_state_only",
      "stats": {
        "bytes": 16263948,
        "sha256": "d55b837f9600e82846e396f1a9384d52e99a17c43ac5a7d53bd14bb24d3e7395"
      },
      "architecture": {
        "training_mode": "template-conditioned diffusion repair",
        "precision": "fp32",
        "frozen_encoder_parameters": 2813714,
        "trainable_diffusion_parameters": 1248145,
        "full_parameters": 4061859,
        "uses_encoder_features": true,
        "branches": [
          "atom_reconstruction",
          "fractional_coordinate_repair",
          "pair_distance_repair",
          "lattice_residual_repair",
          "conservative_update_gates"
        ]
      },
      "training": {
        "phase": "diffusion_generation_v1_prod",
        "steps": 225000,
        "operating_mode": "sampler_aware_best_of_n_with_filtering",
        "science_status": "usable_as_constrained_generator_not_one_shot_oracle"
      },
      "final_eval": {
        "atom_accuracy": 0.99988,
        "coord_improvement_fraction": 0.70544,
        "pair_distance_improvement_fraction": 0.7122,
        "lattice_improvement_fraction": 0.55273,
        "sampler_valid_geometry_rate": 0.66406,
        "valid_geometry_rate": 0.75
      }
    },
    "multimodal_controller": {
      "component": "qwen3_14b_controller_partial_fft",
      "repo_id": "AethronPhantom/Nexa_Mat2",
      "path": "multimodal/controller/nexa_mat_controller_fft_pilot_20260518T234148Z/final_model_merged/model.safetensors",
      "artifact_uri": "hf://AethronPhantom/Nexa_Mat2/multimodal/controller/nexa_mat_controller_fft_pilot_20260518T234148Z/final_model_merged/model.safetensors",
      "base_model": "Qwen/Qwen3-14B",
      "status": "experimental_controller_partial_fft"
    },
    "cross_attention": {
      "component": "nexa_mat_cross_attention_contract_v1",
      "status": "interface_contract",
      "purpose": "condition diffusion proposals and multimodal controller evidence on shared structure, text, task, and model-state lanes",
      "lanes": [
        {
          "name": "structure_latent",
          "producer": "encoder/v1/nexa_mat_V1_final.safetensors",
          "consumer": "diffusion decoder and controller"
        },
        {
          "name": "diffusion_state",
          "producer": "decoder/diffusion_v1/final_checkpoint.safetensors",
          "consumer": "controller ranking and evidence packet"
        },
        {
          "name": "text_intent",
          "producer": "Qwen controller tokens",
          "consumer": "diffusion conditioning and reverse ranking"
        },
        {
          "name": "evidence_context",
          "producer": "retrieval/controller",
          "consumer": "decision layer"
        },
        {
          "name": "task_heads",
          "producer": "SSB/battery-ion adapters",
          "consumer": "reverse pool scoring and use-case constraints"
        }
      ],
      "forward_flow": "constraints -> controller intent tokens -> diffusion proposals -> encoder plausibility scoring -> controller evidence and ranking",
      "reverse_flow": "candidate pool -> encoder/task/evidence scoring -> optional decoder expansion -> controller rank/filter/explain"
    }
  },
  "space": {
    "repo_id": "AethronPhantom/nexamat-crystal-viewer",
    "url": "https://huggingface.co/spaces/AethronPhantom/nexamat-crystal-viewer",
    "runtime": "gradio.Server custom frontend with ZeroGPU model_stack hook"
  },
  "safety_and_limits": [
    "Generated structures are candidates for triage, not DFT-confirmed stable materials.",
    "Decoder is best used as constrained best-of-N generation with encoder filtering.",
    "Controller is experimental and should ground claims in evidence/context rather than inventing support."
  ]
}