1. 10 Oct, 2025 1 commit
  2. 20 Jun, 2025 1 commit
  3. 11 Jun, 2025 1 commit
    • Michael Yang's avatar
      feat: uneven splits (#11048) · 45f56355
      Michael Yang authored
      The current splitDim function only operates on tensors that are split evenly which isn't always the case, e.g. a QKV tensor. This change allows the function to be used for arbitrary splits
      45f56355
  4. 14 May, 2025 1 commit