{
  "model_config_tier": "S",
  "overrides": {
    "d_token": 16,
    "d_model": 32,
    "d_embed": 8,
    "event_layers": 1,
    "event_heads": 2,
    "seq_layers": 1,
    "seq_heads": 2,
    "B_attr": 1024,
    "B_val": 4096,
    "B_etype": 128,
    "num_negatives": 32,
    "max_sampled_tokens": 8
  },
  "trainer": {
    "epochs": 10,
    "global_batch_tokens": 24000,
    "max_tokens_per_microbatch": 16384,
    "checkpoint_every_steps": 20,
    "val_every_steps": 10,
    "compile": false,
    "learning_rate": 0.025
  },
  "seed": 0
}