"""MLX UNet components for HunyuanPaintPBR."""