All pull requests

Pull requests

• Update KernelAbstractions requirement from 0.9.30 to 0.9.30, 0.10

  dependenciesjulia
  #3581·dependabot[bot] opened Oct 25, 2026Bot·

  ·

• Update GPUCompiler requirement from 1.17 to 1.17, 2.16

  dependenciesjulia
  #3580·dependabot[bot] opened Oct 25, 2026Bot·

  ·

• Update LLVM requirement from 9.4.6 to 9.4.6, 10.3

  dependenciesjulia
  #3579·dependabot[bot] opened Oct 25, 2026Bot·

  ·

• Run slice_rotate before recognize_multirotate in the communication passes

  #3578·wsmoses opened Oct 25, 2026Member·

  ·

• Sink the slices feeding a sharded DUS down to the DUS (dus_sink_slices_comm)

  #3576·wsmoses opened Oct 24, 2026Member·

  ·

• Accept types as kernel arguments

  #3571·dkytezab opened Oct 23, 2026Collaborator·

  ·
  1

• Clamp loaded values of CuTracedRNumber arguments in kernels

  #3568·dkytezab opened Oct 23, 2026Collaborator·

  ·
  1

• Cleanup backend provenance in Compiler

  #3566·Pangoraw opened Oct 23, 2026Collaborator·

  ·
  1

  2

• Use polygeist-to-llvm for GPU compilation

  #3565·Pangoraw opened Oct 23, 2026Collaborator·

  ·
  1

• Add in frontend for fixed point implicit diff

  #3561·dkytezab opened Oct 22, 2026Collaborator·

  ·
  6

• Test raising a kernel with a traced Bool argument and Bool stores (#3557)

  #3558·wsmoses opened Oct 22, 2026Member·

  ·

• Update EnzymeAD/Enzyme-JAX to commit 72b7d4c352298a3f79b598b64a1228e09d10a8d2

  dependencies
  #3556·enzymead-bot[bot] opened Oct 22, 2026Bot·

  ·

• custom traced and MLIR types on the Reactant boundary

  #3552·mofeing opened Oct 22, 2026Collaborator·

  ·

• Register CUDA library APIs

  #3551·Pangoraw opened Oct 22, 2026Collaborator·

  ·

• Take the dot_general path for a reshaped, viewed or permuted traced matrix times a host array

  #3547·wsmoses opened Oct 22, 2026Member·

  ·
  1

  1

• Initialize absl logging, default XLA logs to WARNING, make SetLogLevel work

  #3546·wsmoses opened Oct 22, 2026Member·

  ·
  1

• Show a location of a disposed context without reading it

  #3545·wsmoses opened Oct 22, 2026Member·

  ·

• Broadcast a static array with a traced array

  #3544·wsmoses opened Oct 22, 2026Member·

  ·

• fix: do not route static arrays of traced numbers through XLA in dot, norm and map

  #3541·junyixu opened Oct 22, 2026·

  ·
  1

• fix: promote traced Bool operands of + and - to Int

  #3540·junyixu opened Oct 22, 2026·

  ·
  1

  6

• Host-side rounding and math on concrete numbers, for both PJRT and IFRT

  #3532·CarloLucibello opened Oct 21, 2026Contributor·

  ·
  1

  1

• Fix count over traced arrays (e.g. sum of a traced OneHotArray)

  #3531·CarloLucibello opened Oct 21, 2026Contributor·

  ·
  1

  6

• Support widening eltypes in scan_impl!

  #3530·CarloLucibello opened Oct 21, 2026Contributor·

  ·
  1

  2

• Don't write autodiff shadows of non-differentiable arrays

  #3529·CarloLucibello opened Oct 21, 2026Contributor·

  ·
  1

• Materialize wrapped traced arrays in NNlib gather/scatter

  #3528·CarloLucibello opened Oct 21, 2026Contributor·

  ·
  1