I had this implemented in the automated post-install setup but I dropped it upon upgrading to kernel 5.13 as I haven’t experienced any issues since then. If someone can confirm disabling ax resolves the problem indeed, I’d readd it.
The method for Arch should work on Ubuntu as well. Just run update-initramfs -u after adding the file and then reboot.