[MLX] Add options for clearing context#20609

metascroy merged 2 commits into

mainpytorch/executorch:mainfrom

mlx-clear-cxtpytorch/executorch:mlx-clear-cxtCopy head branch name to clipboard