Add chunk-parallel gated delta ops for training#1389

tsato081 wants to merge 2 commits into

ml-explore:mainml-explore/mlx-lm:mainfrom

tsato081:chunked-gdn-trainingtsato081/mlx-lm:chunked-gdn-trainingCopy head branch name to clipboard