the tree animation shader is fine for grass too. you can limit it to animate only upper vertives to not to move the on-ground part at all...