{
  "format": "la-ribo-inference-bundle",
  "format_version": 1,
  "code_repository": "https://github.com/GENTEL-lab/La-Ribo",
  "paper": "https://arxiv.org/abs/2610.12236",
  "checkpoint_format": "pytorch_state_dict",
  "variants": {
    "base": {
      "checkpoint": "weights/la_ribo_base.pt",
      "weights": "ema",
      "inference_config": "config/la-ribo-base.yaml",
      "flow_model": {
        "latent_dim": 16,
        "explicit_backbone_mode": "p_c4prime_n",
        "coarse_dim": 9,
        "relpos_dim": 64,
        "dist_num_bins": 16,
        "dist_min": 0.0,
        "dist_max": 32.0,
        "time_emb_dim": 64,
        "token_dim": 512,
        "pair_dim": 256,
        "cond_dim": 128,
        "num_layers": 12,
        "num_heads": 8,
        "use_qkln": true,
        "transition_expansion": 4,
        "pair_update": {
          "enabled": false
        }
      },
      "autoencoder": {
        "num_nt_types": 4,
        "num_torsions": 10,
        "latent_dim": 16,
        "explicit_backbone": {
          "mode": "p_c4prime_n",
          "zero_center": true
        },
        "decoder_conf": {
          "relpos_dim": 64,
          "dist_num_bins": 16,
          "dist_min": 0.0,
          "dist_max": 32.0,
          "token_dim": 256,
          "pair_dim": 128,
          "cond_dim": 64,
          "num_layers": 6,
          "num_heads": 8,
          "use_qkln": true
        }
      }
    },
    "tri": {
      "checkpoint": "weights/la_ribo_tri.pt",
      "weights": "ema",
      "inference_config": "config/la-ribo-tri.yaml",
      "flow_model": {
        "latent_dim": 16,
        "explicit_backbone_mode": "p_c4prime_n",
        "coarse_dim": 9,
        "relpos_dim": 64,
        "dist_num_bins": 16,
        "dist_min": 0.0,
        "dist_max": 32.0,
        "time_emb_dim": 64,
        "token_dim": 512,
        "pair_dim": 256,
        "cond_dim": 128,
        "num_layers": 12,
        "num_heads": 8,
        "use_qkln": true,
        "transition_expansion": 4,
        "pair_update": {
          "enabled": true,
          "every": 2,
          "use_cuequivariance": true,
          "expansion": 2,
          "token_to_pair": "add"
        }
      },
      "autoencoder": {
        "num_nt_types": 4,
        "num_torsions": 10,
        "latent_dim": 16,
        "explicit_backbone": {
          "mode": "p_c4prime_n",
          "zero_center": true
        },
        "decoder_conf": {
          "relpos_dim": 64,
          "dist_num_bins": 16,
          "dist_min": 0.0,
          "dist_max": 32.0,
          "token_dim": 256,
          "pair_dim": 128,
          "cond_dim": 64,
          "num_layers": 6,
          "num_heads": 8,
          "use_qkln": true
        }
      }
    }
  },
  "shared_files": {
    "autoencoder": "weights/la_ribo_vae.pt",
    "latent_stats": "weights/latent_stats.pt",
    "coarse_stats": "weights/coarse_stats.pt"
  },
  "sampling": {
    "solver": "forward_euler",
    "num_timesteps": 200,
    "scaffold_power": 1.25,
    "latent_power": 2.0,
    "self_condition": true
  },
  "supported_devices": [
    "cpu",
    "cuda"
  ],
  "benchmark_lengths": [
    40,
    50,
    60,
    70,
    80,
    90,
    100,
    110,
    120,
    130,
    140,
    150
  ]
}