All pull requests

Pull requests

• Pass the caller stream through complex l2_norm and scatter max/min VJP

  #4818·Augustas11 opened Nov 2, 2026·1 task done·

  ·

• [docs] Correct the global_scale_x argument name in qqmm

  #4817·Redempt1onzzZZ opened Nov 2, 2026Contributor·

  ·
  1

• [docs] Mark keyword-only parameters in operation signatures

  #4816·Redempt1onzzZZ opened Nov 2, 2026Contributor·

  ·
  1

• Fix missing API documentation and incorrect docstrings

  #4813·XXXXRT666 opened Nov 1, 2026Contributor·

  ·

• Add custom metal kernel for bilinear upsampling

  #4812·starkAhmed43 opened Nov 1, 2026·

  ·
  1

• docs: note that random.normal is not bit-exact across GPU architectures

  #4810·electron-rare opened Oct 31, 2026·

  ·

• Fix row offsets and partition stride in GPU sort

  #4807·ArupKantiDas opened Oct 31, 2026Contributor·

  ·
  1

• [CPU] Fix FP4 tie rounding on CPU

  #4805·404khai opened Oct 30, 2026·

  ·
  1

• [CPU] Fix accumulation precision in affine quantized matmul

  #4803·erwinzhang7 opened Oct 30, 2026Contributor·

  ·
  2

• [Metal] Add affine qqmm with int8 NAX MMAs

  #4802·XXXXRT666 opened Oct 30, 2026Contributor·

  ·
  2

• Security hardening for GitHub Actions

  #4801·madrob opened Oct 30, 2026Contributor·

  ·
  1

• [Metal] Route qqmm without global scales like quantized_matmul

  #4796·XXXXRT666 opened Oct 29, 2026Contributor·

  ·

• Respect MLX_ENABLE_TF32 in gated_delta_update

  #4792·zcbenz opened Oct 29, 2026Member·

  ·
  1

  1

• Validate trace input ids match trace inputs in import_function

  low priority
  #4790·turingH opened Oct 29, 2026Contributor·

  ·

• [Metal] Add affine qmv_mma for 3-16 rows

  #4787·ahuzmeza opened Oct 28, 2026·

  ·
  1

• metal: read FP16 and BF16 affine constants directly for FP32 inputs

  #4785·bri-prism opened Oct 28, 2026Contributor·

  ·
  1

• [CUDA] Sequential kernel for gated_delta_update

  #4783·zcbenz opened Oct 28, 2026Member·

  ·
  1

• [CUDA] Fix gather_mm with sorted_indices=True

  #4771·luciusacarr opened Oct 26, 2026Contributor·

  ·
  1

• Bigger MoE tiles on M1 GPUs

  await response
  #4754·a3rosol opened Oct 23, 2026·

  ·
  1

• Add optional int8 path for quantized matmul with affine weights

  await verification
  #4722·magnuslundstedt opened Oct 18, 2026Contributor·

  ·
  1

• Support uneven tensor parallel sharding in distributed linear layers

  await verificationlow priority
  #4465·twallgren opened Sep 4, 2026·

  ·

• [CUDA] Grouped mm

  await response
  #4390·nastya236 opened Aug 24, 2026Collaborator·

  ·
  1

• Add nanobind 3.0 split-mode support

  await discussion
  #4359·XXXXRT666 opened Aug 19, 2026Contributor·4 tasks done·

  ·
  2

• Add graph-based GPU Timer API

  await discussionlow priority
  #4050·XXXXRT666 opened Aug 7, 2026Contributor·4 tasks done·

  ·

• [CUDA][Improvement] Fast path for gather

  await response
  #3794·nastya236 opened Jul 3, 2026Collaborator·

  ·
  3