Nexa_Mat2 / stack_manifest.json
Allanatrix's picture
chore: publish safetensors NexaMat stack artifacts
3643d86 verified
{
"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."
]
}