some updates to get the accelerated adagrad working