jax

da7b1577 - [mgpu:pallas] Swizzle elements computed using bitwidth rather than bytewidth.