Code

Cleanup for Optimal Control Ops#1045

Code

Merged

jessegrabowski merged 18 commits intopymc-devs:mainpymc-devs/pytensor:mainfrom jessegrabowski:lyapunov-jaxjessegrabowski/pytensor:lyapunov-jaxCopy head branch name to clipboardOct 24, 2024

+301-124

Commits

Commits on Oct 24, 2024

• Blockwise optimal linear control ops

  jessegrabowski

  committed
  57f73e8

  Copy full SHA for 57f73e8

• Add jax rewrite to eliminate BilinearSolveDiscreteLyapunov

  jessegrabowski

  committed
  1674667

  Copy full SHA for 1674667

• set solve_discrete_lyapunov method default to bilinear

  jessegrabowski

  committed
  c14ad65

  Copy full SHA for c14ad65

• Appease mypy

  jessegrabowski

  committed
  10453bb

  Copy full SHA for 10453bb

• restore method dispatching

  jessegrabowski

  committed
  c7057dc

  Copy full SHA for c7057dc

• Use pt.vectorize on base solve_discrete_lyapunov case

  jessegrabowski

  committed
  7e2eaae

  Copy full SHA for 7e2eaae

• Apply JAX rewrite before canonicalization

  jessegrabowski

  committed
  2b751eb

  Copy full SHA for 2b751eb

• Improve tests

  jessegrabowski

  committed
  30759f4

  Copy full SHA for 30759f4

• Remove useless warning filters

  

  ricardoV94

  authored and

  jessegrabowski

  committed
  132f7a5

  Copy full SHA for 132f7a5

• Fix local_blockwise_alloc rewrite

  Show description for a0466e4
  

  ricardoV94

  authored and

  jessegrabowski

  committed
  a0466e4

  Copy full SHA for a0466e4

• Fix float32 tests

  jessegrabowski

  committed
  ebf2d72

  Copy full SHA for ebf2d72

• Test against complex inputs

  jessegrabowski

  committed
  5e07560

  Copy full SHA for 5e07560

• Appease ViPy (Vieira-py type checking)

  jessegrabowski

  committed
  1f2eceb

  Copy full SHA for 1f2eceb

• Remove condition from TensorLike import

  jessegrabowski

  committed
  455338f

  Copy full SHA for 455338f

• Infer dtype from node.outputs.type.dtype

  jessegrabowski

  committed
  beb1bf0

  Copy full SHA for beb1bf0

• Remove unused mypy ignore

  jessegrabowski

  committed
  fb35d92

  Copy full SHA for fb35d92

• Don't manually set dtype of output

  Show description for cb809c1
  jessegrabowski

  committed
  cb809c1

  Copy full SHA for cb809c1

• Set dtype of Op outputs

  jessegrabowski

  committed
  89d5fd0

  Copy full SHA for 89d5fd0