Code

Custom derivative for np.linalg.det#2809

Code

Merged

mattjj merged 7 commits intojax-ml:masterjax-ml/jax:masterfrom dpfau:masterCopy head branch name to clipboardApr 25, 2020

+135

Commits

Commits on Apr 13, 2020

• Add vjp and jvp rules for jnp.linalg.det

  dpfau

  authored
  6cad38b

  Copy full SHA for 6cad38b

• Add tests for new determinant gradients

  dpfau

  authored
  7752168

  Copy full SHA for 7752168

• Replace index_update with concatenate in cofactor_solve

  Show description for 4706a91
  dpfau

  authored
  4706a91

  Copy full SHA for 4706a91

Commits on Apr 23, 2020

• Changes to cofactor_solve so it can be transposed

  Show description for 9f27f39
  dpfau

  authored
  9f27f39

  Copy full SHA for 9f27f39

• Update det grad tests

  Show description for e73c53f
  dpfau

  authored
  e73c53f

  Copy full SHA for e73c53f

Commits on Apr 24, 2020

• Add explanation to docstring for _cofactor_solve

  dpfau

  authored
  99d78b5

  Copy full SHA for 99d78b5

• Fixed comment

  dpfau

  authored
  70f389a

  Copy full SHA for 70f389a