Initial support for MLX. (#23410)