config.json

{
  "in_dim": 10,
  "out_dim": 1,
  "hidden_dim": 128,
  "depth": 4,
  "heads": 8,
  "mlp_ratio": 2,
  "dropout": 0.0,
  "activation": "gelu",
  "include_pos": true,
  "space_dim": 2,
  "latent_multiplier": 2.0,
  "max_latent_steps": 4
}