1713 Commits

Author SHA1 Message Date
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
c720c02d83 Cleaning up repo a bit 2025-09-02 06:45:24 +00:00
Jake Poznanski
56b08d5aa4 Bump version to v0.3.4 for release v0.3.4 2025-08-31 03:12:39 +00:00
Jake Poznanski
f3cdc78b4f Pushing new version 2025-08-31 03:12:30 +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
edd098093b Reverting version changes that broke, vllm 0.10.1 is not good 2025-08-27 18:55:26 +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
27792664bf Transformers version bump needed also 2025-08-27 16:35:51 +00:00
Jake Poznanski
03c7479a17 VLLM version bump 2025-08-27 16:33:37 +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
3eec58012c Docker ignore 2025-08-26 17:52:50 +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
6be12c2e06 Baseline tests for blanks 2025-08-25 22:01:24 +00:00
Jake Poznanski
ad33672781 fix 2025-08-25 21:04:53 +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
c7aa217281 Scripts to run benchmarks better 2025-08-25 20:12:10 +00:00
Jake Poznanski
b16e4051f6 Saving bench results to s3 2025-08-25 19:53:55 +00:00