-record(rmsprop_state, {square_avg :: viva_tensor@tensor:tensor()}).