Use NLL for step size and early termination, randomly adjust gradient
Pickle the full beta array, not just the average