optimize make_lr0_closure