All pull requests

Pull requests

• [CUDA] port Metal rhs gather_qmm optimizations

  #4554·dhiltgen opened Sep 23, 2026Contributor·

  ·

• [BUG][Metal] Deadlock in fence

  #4552·nastya236 opened Sep 23, 2026Collaborator·

  ·

• Contraction docs addition

  low priority
  #4550·simeetnayan81 opened Sep 23, 2026·

  ·

• fix(reduce): support 64-bit indexing for reductions on views past 2^31 and large arrays

  #4549·nicoloangileri opened Sep 23, 2026·

  ·
  1

• Fix empty unsigned sum/prod failing on Metal

  #4546·CodeWithMoin opened Sep 22, 2026·

  ·

• Fixes grouped Conv1d/Conv2d init to use in_channels // groups

  #4544·CodeWithMoin opened Sep 22, 2026·

  ·
  1

• Fix/split data size span

  await verification
  #4540·michalk8 opened Sep 21, 2026Collaborator·

  ·

• Add C++ linear SVM training example with automatic differentiation

  #4537·StarxSky opened Sep 20, 2026·

  ·

• Fix CustomKernel Metal dispatch silently truncating the requested threadgrid

  #4535·pseudobacon opened Sep 19, 2026·

  ·
  1

• Advances file offset during partial reads and adds EOF/signal handling in ParallelFileReader::read #4526

  #4532·Ritabanm opened Sep 18, 2026Contributor·

  ·
  1

• fix(CUDA): Do not throw from ~CudaHandle while the driver is unloading

  #4531·inureyes opened Sep 18, 2026·

  ·

• Report a dead peer in JACCL instead of spinning forever; name the lin…

  low priority
  #4530·Sofille65 opened Sep 18, 2026·

  ·

• Fix col_reduce_longcolumn for negative-stride views

  low priority
  #4529·mateuuszzzzz opened Sep 17, 2026·

  ·
  1

• Fix default.metallib lookup under swift test with Swift Build

  low priority
  #4527·heckj opened Sep 17, 2026Contributor·

  ·
  1

• Clear CUDA errors once reported and raise eval errors in buffer protocol

  low priority
  #4523·aleroot opened Sep 16, 2026Contributor·

  ·

• Add fused Metal kernels for fast.cross_entropy

  await verification
  #4520·zsun6 opened Sep 16, 2026·

  ·
  1

• Fix ordering of complex64 with NaNs

  low priority
  #4519·louen opened Sep 15, 2026Contributor·

  ·
  1

  3

• Improve pre-M5 metal memory usage for SDPA D512

  #4518·dhiltgen opened Sep 15, 2026Contributor·

  ·

• fix(cpu): use size_t for contiguous binary op element counts (#4479)

  low priority
  #4517·Prudctual opened Sep 15, 2026·

  ·
  1

  1

• Use the fast qmv kernel for affine outputs not divisible by 8

  low priority
  #4516·wyanzhao opened Sep 15, 2026Contributor·

  ·
  1

  1

• fix floor_divide

  low priority
  #4515·aaishwarymishra opened Sep 15, 2026Contributor·

  ·
  1

• Fix CUDA termination crash via deterministic worker teardown and lazy stream evaluation

  #4514·aleroot opened Sep 15, 2026Contributor·

  ·
  1

• Clamp slice bounds when a slice is combined with an array index

  low priority
  #4513·IshaanPotle opened Sep 15, 2026·

  ·
  1

• Fix floating-point constant precision in compiled kernels

  low priority
  #4511·Ryan11c opened Sep 15, 2026·

  ·
  1

• Support shapeless compilation of scan operations

  low priority
  #4510·keeeeenw opened Sep 15, 2026·

  ·
  1