Support building on arm64

It would be good to support arm64, this is probably not too painful and can be done by swapping docker build with docker buildx, but there's often edge cases where some code changes are needed.