{
  "settings": {
    "phase": "categorical",
    "train_steps": 400,
    "seeds": 2,
    "sample_steps": [
      32,
      128
    ],
    "samples": 2048,
    "learned_samples": 512,
    "learned_sample_steps": 64,
    "repeats": 3,
    "legacy_categorical": false
  },
  "jax": "0.9.2",
  "python": "3.12.11",
  "devices": [
    "TFRT_CPU_0"
  ],
  "rows": [
    {
      "case": {
        "kind": "linear",
        "steps": 32,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.04443359375,
        0.14404296875,
        0.3115234375,
        0.5
      ],
      "total_variation": 0.011523434892296791,
      "cold_seconds": 0.23731850000331178,
      "warm_ms": 4.492958018090576
    },
    {
      "case": {
        "kind": "linear",
        "steps": 32,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.04443359375,
        0.14404296875,
        0.3115234375,
        0.5
      ],
      "total_variation": 0.011523434892296791,
      "cold_seconds": 0.20443266700021923,
      "warm_ms": 4.71820798702538
    },
    {
      "case": {
        "kind": "linear",
        "steps": 128,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.05029296875,
        0.1376953125,
        0.2978515625,
        0.51416015625
      ],
      "total_variation": 0.014453133568167686,
      "cold_seconds": 0.22743570804595947,
      "warm_ms": 14.057791966479272
    },
    {
      "case": {
        "kind": "linear",
        "steps": 128,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.05029296875,
        0.1376953125,
        0.2978515625,
        0.51416015625
      ],
      "total_variation": 0.014453133568167686,
      "cold_seconds": 0.24322595802368596,
      "warm_ms": 13.79937498131767
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 32,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "frequencies": [
        0.05126953125,
        0.150390625,
        0.29248046875,
        0.505859375
      ],
      "total_variation": 0.007519533857703209,
      "cold_seconds": 0.22054462495725602,
      "warm_ms": 4.182000004220754
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 32,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "frequencies": [
        0.05126953125,
        0.150390625,
        0.29248046875,
        0.505859375
      ],
      "total_variation": 0.007519533857703209,
      "cold_seconds": 0.21840195800177753,
      "warm_ms": 4.522915987763554
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 128,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "frequencies": [
        0.0498046875,
        0.16064453125,
        0.30517578125,
        0.484375
      ],
      "total_variation": 0.015820303931832314,
      "cold_seconds": 0.22968233400024474,
      "warm_ms": 13.920917001087219
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 128,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "frequencies": [
        0.0498046875,
        0.16064453125,
        0.30517578125,
        0.484375
      ],
      "total_variation": 0.015820303931832314,
      "cold_seconds": 0.23561158397933468,
      "warm_ms": 13.562749954871833
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 32,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.04541015625,
        0.14306640625,
        0.3076171875,
        0.50390625
      ],
      "total_variation": 0.011523434892296791,
      "cold_seconds": 0.2362432919908315,
      "warm_ms": 3.7537500029429793
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 32,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.04541015625,
        0.14306640625,
        0.3076171875,
        0.50390625
      ],
      "total_variation": 0.011523434892296791,
      "cold_seconds": 0.23264416598249227,
      "warm_ms": 4.399375000502914
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 128,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.04833984375,
        0.150390625,
        0.28515625,
        0.51611328125
      ],
      "total_variation": 0.01650390960276127,
      "cold_seconds": 0.22977770800935104,
      "warm_ms": 13.99116701213643
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 128,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "frequencies": [
        0.04833984375,
        0.150390625,
        0.28515625,
        0.51611328125
      ],
      "total_variation": 0.01650390960276127,
      "cold_seconds": 0.27528641698881984,
      "warm_ms": 14.143000007607043
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 32,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "frequencies": [
        0.04541015625,
        0.15966796875,
        0.298828125,
        0.49609375
      ],
      "total_variation": 0.009667972102761269,
      "cold_seconds": 0.20401212497381493,
      "warm_ms": 4.48416598374024
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 32,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "frequencies": [
        0.04541015625,
        0.15966796875,
        0.298828125,
        0.49609375
      ],
      "total_variation": 0.009667972102761269,
      "cold_seconds": 0.21451320900814608,
      "warm_ms": 4.604666959494352
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 128,
        "oracle": true,
        "matched_terminal": false
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "frequencies": [
        0.0576171875,
        0.158203125,
        0.30615234375,
        0.47802734375
      ],
      "total_variation": 0.021972646936774254,
      "cold_seconds": 0.22468879102962092,
      "warm_ms": 13.618291995953768
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 128,
        "oracle": true,
        "matched_terminal": true
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "frequencies": [
        0.0576171875,
        0.158203125,
        0.30615234375,
        0.47802734375
      ],
      "total_variation": 0.021972646936774254,
      "cold_seconds": 0.22621100000105798,
      "warm_ms": 14.35804198263213
    },
    {
      "case": {
        "kind": "linear",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.42750425002304837,
      "finite_losses": true,
      "final_loss": 0.9249849319458008,
      "validation_ce": 0.9846087694168091,
      "bayes_ce": 0.9745603203773499,
      "excess_ce": 0.010048449039459229,
      "frequencies": [
        0.10302734375,
        0.17724609375,
        0.2822265625,
        0.4375
      ],
      "total_variation": 0.08027344010770321,
      "cold_seconds": 0.2607291669701226,
      "warm_ms": 15.230457996949553
    },
    {
      "case": {
        "kind": "linear",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.41526287503074855,
      "finite_losses": true,
      "final_loss": 0.8595866560935974,
      "validation_ce": 0.9355015754699707,
      "bayes_ce": 0.9293696880340576,
      "excess_ce": 0.006131887435913086,
      "frequencies": [
        0.1201171875,
        0.1865234375,
        0.26806640625,
        0.42529296875
      ],
      "total_variation": 0.10664062760770321,
      "cold_seconds": 0.25144354201620445,
      "warm_ms": 15.093749971129
    },
    {
      "case": {
        "kind": "linear",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.5035582080017775,
      "finite_losses": true,
      "final_loss": 1.0589087009429932,
      "validation_ce": 0.9868855476379395,
      "bayes_ce": 0.9745603203773499,
      "excess_ce": 0.0123252272605896,
      "frequencies": [
        0.13720703125,
        0.1982421875,
        0.2724609375,
        0.39208984375
      ],
      "total_variation": 0.1354492213577032,
      "cold_seconds": 0.26797687500948086,
      "warm_ms": 13.631667010486126
    },
    {
      "case": {
        "kind": "linear",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.5373757500201464,
      "finite_losses": true,
      "final_loss": 1.0422395467758179,
      "validation_ce": 0.9465026259422302,
      "bayes_ce": 0.9293696880340576,
      "excess_ce": 0.017132937908172607,
      "frequencies": [
        0.16064453125,
        0.19970703125,
        0.25927734375,
        0.38037109375
      ],
      "total_variation": 0.1603515651077032,
      "cold_seconds": 0.25723462499445304,
      "warm_ms": 15.01320896204561
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "training_seconds": 0.45582708303118125,
      "finite_losses": true,
      "final_loss": 0.6544968485832214,
      "validation_ce": 0.8264967799186707,
      "bayes_ce": 0.8221867680549622,
      "excess_ce": 0.004310011863708496,
      "frequencies": [
        0.08984375,
        0.1669921875,
        0.287109375,
        0.4560546875
      ],
      "total_variation": 0.05683594010770321,
      "cold_seconds": 0.24216995801543817,
      "warm_ms": 13.656041002832353
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "training_seconds": 0.45330166700296104,
      "finite_losses": true,
      "final_loss": 0.6810919046401978,
      "validation_ce": 0.7642863988876343,
      "bayes_ce": 0.7602081894874573,
      "excess_ce": 0.004078209400177002,
      "frequencies": [
        0.08251953125,
        0.162109375,
        0.28662109375,
        0.46875
      ],
      "total_variation": 0.04462890885770321,
      "cold_seconds": 0.2511432499741204,
      "warm_ms": 13.246457965578884
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "training_seconds": 0.5086986670503393,
      "finite_losses": true,
      "final_loss": 1.0122562646865845,
      "validation_ce": 0.8320434093475342,
      "bayes_ce": 0.8221867680549622,
      "excess_ce": 0.009856641292572021,
      "frequencies": [
        0.1328125,
        0.19189453125,
        0.27587890625,
        0.3994140625
      ],
      "total_variation": 0.12470703385770321,
      "cold_seconds": 0.24529562500538304,
      "warm_ms": 13.637458032462746
    },
    {
      "case": {
        "kind": "cosine",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 9.999999960041972e-13,
      "training_seconds": 0.5537998329964466,
      "finite_losses": true,
      "final_loss": 1.032496452331543,
      "validation_ce": 0.780543327331543,
      "bayes_ce": 0.7602081894874573,
      "excess_ce": 0.020335137844085693,
      "frequencies": [
        0.14111328125,
        0.19482421875,
        0.28076171875,
        0.38330078125
      ],
      "total_variation": 0.1359375026077032,
      "cold_seconds": 0.24854987498838454,
      "warm_ms": 13.086792023386806
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.457488666055724,
      "finite_losses": true,
      "final_loss": 0.90073162317276,
      "validation_ce": 0.9779899716377258,
      "bayes_ce": 0.9686376452445984,
      "excess_ce": 0.009352326393127441,
      "frequencies": [
        0.09716796875,
        0.17578125,
        0.279296875,
        0.44775390625
      ],
      "total_variation": 0.07294922135770321,
      "cold_seconds": 0.2717018330004066,
      "warm_ms": 13.410374987870455
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.47087575000477955,
      "finite_losses": true,
      "final_loss": 0.8423827886581421,
      "validation_ce": 0.9243450164794922,
      "bayes_ce": 0.9198939800262451,
      "excess_ce": 0.00445103645324707,
      "frequencies": [
        0.11572265625,
        0.17822265625,
        0.27294921875,
        0.43310546875
      ],
      "total_variation": 0.09394531510770321,
      "cold_seconds": 0.26528045802842826,
      "warm_ms": 13.804916990920901
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.48172766697825864,
      "finite_losses": true,
      "final_loss": 1.0589210987091064,
      "validation_ce": 0.9813955426216125,
      "bayes_ce": 0.9686376452445984,
      "excess_ce": 0.01275789737701416,
      "frequencies": [
        0.1328125,
        0.20361328125,
        0.27978515625,
        0.3837890625
      ],
      "total_variation": 0.1364257838577032,
      "cold_seconds": 0.26036700000986457,
      "warm_ms": 16.109791002236307
    },
    {
      "case": {
        "kind": "sigmoid",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 4.318574065109715e-05,
      "training_seconds": 0.4984682910144329,
      "finite_losses": true,
      "final_loss": 1.0422323942184448,
      "validation_ce": 0.9384788870811462,
      "bayes_ce": 0.9198939800262451,
      "excess_ce": 0.018584907054901123,
      "frequencies": [
        0.162109375,
        0.19970703125,
        0.26416015625,
        0.3740234375
      ],
      "total_variation": 0.1618164088577032,
      "cold_seconds": 0.255232666968368,
      "warm_ms": 14.87862499197945
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "training_seconds": 0.45715475000906736,
      "finite_losses": true,
      "final_loss": 0.567887544631958,
      "validation_ce": 0.6075151562690735,
      "bayes_ce": 0.6059514880180359,
      "excess_ce": 0.0015636682510375977,
      "frequencies": [
        0.09375,
        0.1708984375,
        0.28515625,
        0.4501953125
      ],
      "total_variation": 0.06464844010770321,
      "cold_seconds": 0.25831475004088134,
      "warm_ms": 14.817374991253018
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 64,
        "oracle": false,
        "training": "uniform",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "training_seconds": 0.4455764160375111,
      "finite_losses": true,
      "final_loss": 0.5382218360900879,
      "validation_ce": 0.55897456407547,
      "bayes_ce": 0.5578790307044983,
      "excess_ce": 0.0010955333709716797,
      "frequencies": [
        0.099609375,
        0.17333984375,
        0.27880859375,
        0.4482421875
      ],
      "total_variation": 0.07294922135770321,
      "cold_seconds": 0.24788533296668902,
      "warm_ms": 14.436499972362071
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 0
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "training_seconds": 0.5100221250322647,
      "finite_losses": true,
      "final_loss": 1.0135196447372437,
      "validation_ce": 0.6092057228088379,
      "bayes_ce": 0.6059514880180359,
      "excess_ce": 0.003254234790802002,
      "frequencies": [
        0.1171875,
        0.177734375,
        0.2802734375,
        0.4248046875
      ],
      "total_variation": 0.09492187760770321,
      "cold_seconds": 0.25106412498280406,
      "warm_ms": 14.57466697320342
    },
    {
      "case": {
        "kind": "logsnr",
        "steps": 64,
        "oracle": false,
        "training": "importance",
        "seed": 1
      },
      "status": "ok",
      "terminal_alpha_bar": 0.0024726230185478926,
      "training_seconds": 0.5338687499752268,
      "finite_losses": true,
      "final_loss": 0.9987914562225342,
      "validation_ce": 0.5639774799346924,
      "bayes_ce": 0.5578790307044983,
      "excess_ce": 0.006098449230194092,
      "frequencies": [
        0.138671875,
        0.19384765625,
        0.27392578125,
        0.3935546875
      ],
      "total_variation": 0.1325195338577032,
      "cold_seconds": 0.2508681660401635,
      "warm_ms": 14.288624981418252
    }
  ]
}