mlx: use aligned memory to register regions