rntn gradient fixes