flax/processor/bit.hpp: mask Add