1603 Commits

Author SHA1 Message Date
Jake Poznanski
c0a1e70440 Adding filter 2025-09-08 17:02:09 +00:00
Jake Poznanski
aad8729ad7 prepare checkpint from main 2025-09-08 17:01:58 +00:00
Jake Poznanski
0f46fa0988 Adding num_generations 2025-09-04 18:44:23 +00:00
Jake Poznanski
ef09c73bf2 Fixing up some rewards stuff 2025-09-04 17:34:53 +00:00
Jake Poznanski
ede0dc51b1 Adding drop last to prevent any weirdnesses 2025-09-04 16:50:08 +00:00
Jake Poznanski
14a882db9a Fixing to new version, adjusting scale rewards stuff 2025-09-03 22:43:35 +00:00
Jake Poznanski
2fd4ae8489 Adding some more options to play with 2025-09-03 22:29:23 +00:00
Jake Poznanski
755c221024 Trying some more things 2025-09-03 22:11:16 +00:00
Jake Poznanski
0a9c8f3e96 Adding warmup steps param 2025-09-03 21:33:18 +00:00
Jake Poznanski
a41d04660a Cleaning script 2025-09-03 21:31:21 +00:00
Jake Poznanski
e6cff25b6b Cleanup stuff 2025-09-03 20:34:12 +00:00
Jake Poznanski
bade86fe91 Cleaned up things 2025-09-03 20:23:01 +00:00
Jake Poznanski
b689a8e5f8 Giving more memory buffer 2025-09-03 19:56:53 +00:00
Jake Poznanski
7346d12322 Better cleaning, augusta version 2025-09-03 18:47:02 +00:00
Jake Poznanski
f20f1a0b54 Doing some cleaning 2025-09-03 18:41:36 +00:00
Jake Poznanski
94d19c51c6 Cleaning up scripts, multi gpu trainer more flexible 2025-09-03 18:25:10 +00:00
Jake Poznanski
c612293a59 Remove device map auto 2025-09-03 18:04:42 +00:00
Jake Poznanski
1fb49cefc1 Working on multi gpu trainer 2025-09-03 17:25:14 +00:00
Jake Poznanski
00f51fb2c7 Fixing bug with multi epoch training 2025-09-02 21:03:00 +00:00
Jake Poznanski
72fcfafde7 Fixed up national archives script 2025-08-29 16:36:18 +00:00
Jake Poznanski
7e09a02f7c Better NA extractor 2025-08-28 20:48:51 +00:00
Jake Poznanski
22626a512c Describing national archives project 2025-08-28 18:44:24 +00:00
Jake Poznanski
f426826850 Clean downloads 2025-08-28 18:03:52 +00:00
Jake Poznanski
3d3d184f25 Fixes 2025-08-28 17:56:27 +00:00
Jake Poznanski
ed3820c0c7 LOC downloader 2025-08-28 17:38:20 +00:00
Jake Poznanski
6123d4452b Reorganizing files 2025-08-28 17:04:26 +00:00
Jake Poznanski
0710debf75 Cleaner front matter reward 2025-08-27 19:49:42 +00:00
Jake Poznanski
09036b07d9 Verifying bench loading 2025-08-27 19:00:29 +00:00
Jake Poznanski
14f19e5d58 Some cost tracking 2025-08-27 18:53:42 +00:00
Jake Poznanski
d70208d98a Moving test code around, adding format reward since some runs stop outputting the front matter thing in grpo training 2025-08-27 18:22:05 +00:00
Jake Poznanski
8383865392 Fixing up subscripts and superscripts in synth data 2025-08-27 18:15:36 +00:00
Jake Poznanski
cf479d1d29 Fixing reward weights stuff 2025-08-26 22:04:45 +00:00
Jake Poznanski
c86b413d3e Adding additional rewards and weights 2025-08-26 21:49:35 +00:00
Jake Poznanski
9c520498dd Adding some losses 2025-08-26 21:18:24 +00:00
Jake Poznanski
3be381b375 Adding some params 2025-08-26 20:46:06 +00:00
Jake Poznanski
4b3660debd Quieting some messages 2025-08-26 20:15:39 +00:00
Jake Poznanski
8327da2415 Fix 2025-08-26 20:06:52 +00:00
Jake Poznanski
98f4d62d1e Fixing data check 2025-08-26 19:51:54 +00:00
Jake Poznanski
3433c8f5f2 Adding more cmd line args 2025-08-26 18:28:41 +00:00
Jake Poznanski
9671f6847c Parallel rewards 2025-08-26 18:16:21 +00:00
Jake Poznanski
2588f911dc Docker ignore 2025-08-26 17:52:30 +00:00
Jake Poznanski
82fd50263f Launcher for grpo training 2025-08-26 16:28:38 +00:00
Jake Poznanski
ba226cc1d5 Saving url metadata 2025-08-25 20:47:54 +00:00
Jake Poznanski
90a7443b2b Working to cleanup miner script 2025-08-25 20:41:04 +00:00
Jake Poznanski
ed6f483074 Fixing run_benchmark 2025-08-25 20:28:40 +00:00
Jake Poznanski
d84eb95ba2 Saving some extra data mixes 2025-08-25 20:26:29 +00:00
Jake Poznanski
b16e4051f6 Saving bench results to s3 2025-08-25 19:53:55 +00:00
Jake Poznanski
d9b6978499 Some scripts 2025-08-25 18:44:18 +00:00
Jake Poznanski
2415cdcfda Adding support for a blank page output check 2025-08-25 17:20:47 +00:00
Jake Poznanski
33ecc2316c Startng miner for blank pages, we want to make sure we don't hallucinate 2025-08-25 16:54:23 +00:00