There's a bug with axolotls. When they swim around and see a big dripleaf 1 block above the water, they start to try and jump up to it (as seen in the video). I think this has to do with their pathfinding strategies, but that's just me.