Switch back to mklabs/tiny-lr