Code

lax.custom_linear_solve primitive#1402

Code

Merged

shoyer merged 29 commits intojax-ml:masterjax-ml/jax:masterfrom shoyer:linear-solversshoyer/jax:linear-solversCopy head branch name to clipboardOct 21, 2019

Commits

Commits on Sep 25, 2019

• WIP: linear solvers

  shoyer

  committed
  6888e4f

  Copy full SHA for 6888e4f

Commits on Sep 26, 2019

• Merge branch 'solvers' into linear-solvers

  shoyer

  committed
  385e282

  Copy full SHA for 385e282

• Draft of lax.linear_solve

  shoyer

  committed
  189ca8c

  Copy full SHA for 189ca8c

Commits on Sep 28, 2019

• Merge branch 'master' into linear-solvers

  shoyer

  committed
  2479314

  Copy full SHA for 2479314

• Refactor pytree munging inside lax.root.

  Show description for 5042dde
  shoyer

  committed
  5042dde

  Copy full SHA for 5042dde

• Merge branch 'refactor-root' into linear-solvers

  shoyer

  committed
  42e09e5

  Copy full SHA for 42e09e5

Commits on Sep 29, 2019

• Fixup linear_solve

  shoyer

  committed
  d3211a2

  Copy full SHA for d3211a2

• Linearize multiple times in _root_jvp to avoid zeros

  shoyer

  committed
  2281105

  Copy full SHA for 2281105

• Merge branch 'refactor-root' into linear-solvers

  shoyer

  committed
  c9cfdf0

  Copy full SHA for c9cfdf0

• fix deftraced

  shoyer

  committed
  8a2ec33

  Copy full SHA for 8a2ec33

• add a symmetric argument

  shoyer

  committed
  b1806c4

  Copy full SHA for b1806c4

Commits on Oct 2, 2019

• Fixup float64; add a test for symmetric/non-symmetric

  shoyer

  committed
  5c9c8cc

  Copy full SHA for 5c9c8cc

• test zeros in linear_solve_jvp

  shoyer

  committed
  3e39769

  Copy full SHA for 3e39769

Commits on Oct 3, 2019

• Revisions per review

  shoyer

  committed
  d5aaaad

  Copy full SHA for d5aaaad

• Merge branch 'refactor-root' into linear-solvers

  shoyer

  committed
  39c3f5c

  Copy full SHA for 39c3f5c

• Adjust signature of linear_solve

  shoyer

  committed
  f5f3007

  Copy full SHA for f5f3007

• Merge branch 'master' into linear-solvers

  shoyer

  committed
  709899b

  Copy full SHA for 709899b

• restore botched test

  shoyer

  committed
  f98c1b3

  Copy full SHA for f98c1b3

• variable names

  shoyer

  committed
  57cc313

  Copy full SHA for 57cc313

Commits on Oct 10, 2019

• jaxprize

  shoyer

  committed
  fab4f45

  Copy full SHA for fab4f45

• spelling

  shoyer

  committed
  1de6200

  Copy full SHA for 1de6200

• Use np.dot instead of @

  shoyer

  committed
  8fa4ac6

  Copy full SHA for 8fa4ac6

• linear_solve docstring, more tests

  shoyer

  committed
  2e0ccf7

  Copy full SHA for 2e0ccf7

• Disable batching for root and linear_solve

  shoyer

  committed
  0e06627

  Copy full SHA for 0e06627

• Fix linear_solve tests

  shoyer

  committed
  eadb731

  Copy full SHA for eadb731

• remove unused imports

  shoyer

  committed
  edea7ce

  Copy full SHA for edea7ce

• Rename to custom_linear_solve

  shoyer

  committed
  a276348

  Copy full SHA for a276348

• WIP: refactor

  shoyer

  committed
  30c2dc2

  Copy full SHA for 30c2dc2

• fixup test for lazy transpose_solve error

  shoyer

  committed
  20169dc

  Copy full SHA for 20169dc