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 14:39:52 +08:00
2026-07-01 15:42:27 +00:00
2026-07-02 13:00:41 +08:00
2026-06-23 13:11:40 -05:00
2026-03-12 07:57:47 -07:00
2026-02-18 12:19:11 +08:00
2026-07-02 18:48:57 +08:00
2026-06-23 23:33:17 +00:00
2026-07-01 09:14:00 +00:00
2026-06-30 13:34:25 -04:00
2026-06-12 13:48:15 -07:00
2026-05-15 15:56:13 +00:00
2026-07-02 12:32:05 -07:00
2026-07-01 15:36:13 +08:00
2026-06-30 11:20:33 +00:00
2026-07-02 18:09:27 -07:00
2026-06-30 20:43:42 +00:00
2026-07-01 01:46:01 +00:00
2026-06-16 01:32:32 +00:00
2026-06-17 06:30:49 +00:00
2023-12-13 12:28:13 -08:00
2026-06-29 09:29:39 -07:00
2026-06-30 07:53:17 +00:00
2026-06-30 23:08:51 +08:00
2026-06-24 22:29:19 +08:00
2026-07-01 07:07:06 +00:00
2026-05-13 16:53:01 +00:00
2026-06-23 15:44:43 +00:00
2025-01-01 21:56:46 +09:00
2026-06-25 23:02:10 -07:00
2026-07-02 12:08:37 -07:00
2026-06-30 13:46:53 -04:00
2026-07-02 17:45:23 -07:00
2026-02-16 12:57:39 +00:00
2026-06-26 23:56:39 +00:00
2026-06-30 14:39:52 +08:00
2026-07-03 06:22:59 +00:00
2025-10-12 09:51:31 -07:00
2026-05-15 08:25:52 +08:00
2023-10-31 15:39:38 -07:00
2025-10-13 16:44:50 +08:00
2026-06-29 07:59:24 +00:00
2026-01-26 21:49:03 +00:00
2026-01-09 13:10:24 -08:00
2026-07-02 18:48:57 +08:00
2025-10-05 07:06:22 -07:00
2026-06-11 12:49:08 +08:00
2026-06-16 15:16:45 +02:00
2026-04-10 08:29:35 +00:00
2026-02-12 22:48:38 -08:00
2026-06-23 23:33:17 +00:00
2026-06-04 06:49:31 +00:00
2025-11-19 02:10:02 +00:00
2025-10-05 07:06:22 -07:00
2026-06-22 10:16:35 +00:00
2026-04-01 14:34:29 -07:00
2026-05-11 07:40:37 +00:00
2026-05-09 19:47:09 -07:00
2025-10-05 07:06:22 -07:00
2025-10-05 07:06:22 -07:00
2025-10-05 07:06:22 -07:00
2025-10-05 07:06:22 -07:00
2025-10-05 07:06:22 -07:00
2026-03-15 23:35:35 +00:00
2026-06-30 15:27:21 +08:00