I think speed can be improved further, by constructing a parallel version. Possibly, this would be similar to how BLAKE3 used Bao mode to parallelize.