max |W_scratch - W_altform| : 2.220446049250313e-16
max |b_scratch - b_altform| : 0.0
loss[0] -> loss[-1] (scratch): 4.6282430304594495 -> 0.03158175143063779
loss trajectories match: True
hand-derived theta_1  : 0.9899000002
AdamWScratch theta_1  : 0.9899000002
