Better loss grads pluggability
parmaeter changes
better stepsizes and library of proxes