All pull requests

Pull requests

• Speedup numba basic set_subtensor by using direct update loop

  #2574·ricardoV94 opened Oct 27, 2026Member·

  ·

• InplaceOptimizer: Don't retry the optimistic strategy that was rejected already

  #2573·ricardoV94 opened Oct 27, 2026Member·

  ·

• Drop cached pullback and pushforward when pickling OpFromGraph

  #2572·ShryukGrandhi opened Oct 26, 2026·5 of 11 tasks·

  ·
  1

• Fix geomspace endpoint signs for negative start and stop

  #2570·mayuriphad opened Oct 26, 2026·5 of 11 tasks·

  ·
  1

  1

• Fix PyTorch Prod accumulator and output dtypes

  #2569·xgbtxy opened Oct 26, 2026·5 tasks done·

  ·
  1

• fix: return Python ints from as_index_literal for NumPy scalar constants

  #2568·nosmile99 opened Oct 25, 2026·5 of 6 tasks·

  ·
  1

• Preserve input dtype for constant and edge padding

  #2566·AryanRG opened Oct 25, 2026·5 tasks done·

  ·

• split pytensor-cache introspection into collect and render steps

  #2564·stephenyang0215 opened Oct 25, 2026·7 of 11 tasks·

  ·
  1

  2

• fix: PyTorch Sum ignoring output/accumulator dtype and reducing axis=()

  #2562·stephenyang0215 opened Oct 25, 2026·6 of 11 tasks·

  ·
  1

• Keep sparse-sparse multiply in the declared format

  #2560·Souravrajvi0 opened Oct 24, 2026·5 of 6 tasks·

  ·
  1

• Accept bare Python scalars from bracketing root_scalar methods

  #2557·jessegrabowski opened Oct 23, 2026Member·

  ·
  5

• Rewrite knowable logical comparisons

  graph rewriting
  #2556·ricardoV94 opened Oct 22, 2026Member·

  ·
  1

• Add jacobian_sparsity with Elemwise sparsity rules

  gradientsnew feature
  #2545·jessegrabowski opened Oct 22, 2026Member·

  ·
  3

• Add nan-unsafe rewrite for ge(exp(x), 0)

  #2528·alisha-1000 opened Oct 20, 2026·

  ·

• Fix empty dimensions in batched matmul optimizations

  #2527·galleonli opened Oct 20, 2026·5 of 11 tasks·

  ·
  1

  11

• Make bincount match NumPy for small integers, weights, bool and empty input

  #2522·spicysharkco opened Oct 19, 2026·5 of 11 tasks·

  ·
  2

• Fix argmin and min of integers at the extremes of the dtype range

  #2515·spicysharkco opened Oct 19, 2026·5 of 11 tasks·

  ·
  6

• Store constant reversed slice steps directly in Subtensor idx_list

  graph rewritingindexingmajorrefactor
  #2513·jessegrabowski opened Oct 18, 2026Member·

  ·
  1

  18

• Move is_provably_positive and is_provably_non_negative to tensor.predicate

  #2512·ADITYASINGH1206 opened Oct 18, 2026Contributor·

  ·
  1

  11

• Fix vectorizing random variables whose size comes from a batched shape

  #2510·Cherith1222 opened Oct 18, 2026·5 of 11 tasks·

  ·
  1

  2

• Add a symbolic cost model for graphs and ops

  graph objectsgraph rewritinglinalgnew feature
  #2508·jessegrabowski opened Oct 18, 2026Member·

  ·
  1

• Allow direct scalar inputs in JAX Blockwise

  bugjaxscanvectorization
  #2498·jessegrabowski opened Oct 14, 2026Member·

  ·
  1

  13

• Bump the github-actions group across 1 directory with 12 updates

  GitHub CI/CDmajorno releasenotes
  #2497·dependabot[bot] opened Oct 14, 2026Bot·

  ·

• Support len on tensors with a known first axis length

  #2494·err0rgod opened Oct 13, 2026·7 of 11 tasks·

  ·
  1

  3

• Implement native Numba BFGS minimization

  #2492·ricardoV94 opened Oct 13, 2026Member·

  ·
  4