Code

Fixed point iterator#978

Code

Draft

theorashid wants to merge 7 commits intopymc-devs:mainpymc-devs/pytensor:mainfrom theorashid:fixed-point-iteratortheorashid/pytensor:fixed-point-iteratorCopy head branch name to clipboard

+217

Commits

Commits on Aug 19, 2024

• feat: Add fixed point solver for optimization, with tests comparing to jax

  Show description for 3d006ff
  

  5 peoplecommitted
  3d006ff

  Copy full SHA for 3d006ff

• Further information on where test is from

  theorashid

  committed
  ff5c32f

  Copy full SHA for ff5c32f

• remove commented test case

  theorashid

  committed
  01858c4

  Copy full SHA for 01858c4

• remove debug print statements from scan

  theorashid

  committed
  c2f2fce

  Copy full SHA for c2f2fce

• use trace to work out n_steps (see https://github.com/aseyboldt/pytensor/blob/newton/experiment-newton.ipynb)

  theorashid

  committed
  0121ec3

  Copy full SHA for 0121ec3

• add TODO refactor newton solver to use fwd_solver

  theorashid

  committed
  a49ade3

  Copy full SHA for a49ade3

Commits on Sep 11, 2024

• add notes so hackers can see

  theorashid

  committed
  57aaf8c

  Copy full SHA for 57aaf8c