gradient on mu working for un-transformed case