[MLX] Add the cell KV-cache layout to the MLX runtime - #22167

[MLX] Add the cell KV-cache layout to the MLX runtime#22167

kiymetakdemir merged 3 commits into

pytorch:mainpytorch/executorch:mainfrom

kiymetakdemir:mlx-cell-cachekiymetakdemir/executorch:mlx-cell-cacheCopy head branch name to clipboard