Woosuk Kwon and Claude
50cfbbca7a
[MRV2] Draw 64-bit uniforms for fp32 Gumbel sampling
...
tl.rand only resolves the uniform down to 2**-31, which hard-caps the
fp32 Gumbel noise at ~21.5 and coarsely buckets the argmax-deciding
tail. Philox already generates 128 bits per call (tl.rand discards 96),
so draw 64 bits instead and convert uint64 -> fp32 directly: the tail
now resolves down to 2**-64 (noise up to ~44.4) with no fp64 arithmetic
and no measurable cost. The fp64 path is unchanged.
Co-authored-by: Claude
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai >
2026-07-03 06:22:59 +00:00
..
2026-06-30 20:43:42 +00:00
2026-07-01 15:42:27 +00:00
2026-07-02 13:00:41 +08:00
2026-07-02 18:48:57 +08:00
2026-06-29 18:34:18 -07:00
2026-07-01 07:17:44 +00:00
2026-07-01 01:25:19 -07:00
2026-07-01 01:25:19 -07:00
2026-07-01 10:00:19 +00:00
2026-06-28 00:07:33 -07:00
2026-06-05 14:51:44 +00:00
2026-06-29 19:03:06 +00:00
2026-03-31 22:15:05 -04:00
2026-07-01 22:14:04 +08:00
2026-07-02 19:57:41 -07:00
2026-07-01 21:38:51 -07:00
2026-06-30 09:39:22 +00:00
2026-07-01 01:46:01 +00:00
2026-06-30 23:24:40 -07:00
2026-05-29 14:04:58 +00:00
2026-04-22 16:42:43 -04:00
2026-02-17 01:08:42 -08:00
2026-07-01 10:16:07 +00:00
2026-06-30 23:08:51 +08:00
2026-01-21 22:02:39 -07:00
2026-06-27 23:09:18 -07:00
2026-07-02 12:08:37 -07:00
2026-02-05 16:59:28 -08:00
2026-07-01 17:32:27 -07:00
2026-06-24 08:41:56 +08:00
2026-03-31 22:32:54 +08:00
2026-07-02 12:32:05 -07:00
2026-07-03 06:22:59 +00:00
2026-06-30 12:17:34 -07:00
2026-02-26 03:01:01 -08:00
2026-07-02 13:00:41 +08:00
2026-07-02 12:31:26 -07:00
2026-06-12 11:45:01 +00:00
2026-04-13 12:51:46 +08:00
2026-03-19 19:00:20 +00:00
2026-06-16 15:16:45 +02:00
2026-06-30 09:39:22 +00:00
2026-03-10 20:16:30 -07:00
2026-06-23 14:54:46 -07:00
2026-03-19 12:04:59 -07:00
2026-01-29 09:12:26 +00:00
2026-01-09 10:12:35 -07:00
2026-05-13 23:49:01 -07:00
2026-06-22 10:16:35 +00:00
2026-06-30 22:41:33 +00:00
2026-06-24 13:53:54 -04:00
2026-05-28 13:12:03 -07:00
2026-04-21 12:27:44 +00:00