Interface DflashBatch

interface DflashBatch {
    anchorToks: number[];
    blockToks: number[][];
    ctxMask: MlxArray;
    hCtx: MlxArray;
    size: number;
    targetHidden: MlxArray;
}