google-deepmind/kfac-jax

Issues

• Can this be used for Laplace approximation?

  #224 opened 6 months ago by blackblitz
  1

• ImportError: cannot import name 'psd_inv_cholesky' from 'kfac_jax._src.utils'

  #256 opened 3 months ago by eul8
  1

• Compatibility with `pallas` attention

  #244 opened 4 months ago by ae-foster
  10

• Add Support for KFAC Optimization in LSTM and GRU Layers

  #188 opened a year ago by neuronphysics
  4

• Using K-FAC with physics-based losses

  #186 opened a year ago by SNMS95
  3

• Value functions returning state objects not supported

  #131 opened a year ago by rmclarke
  4

• KFAC Norm Constraint

  #160 opened a year ago by Baiyu-Su
  2

• Using kfac inside jitted function

  #103 opened a year ago by Honza9723
  2

• Quickstart example with different NN libraries does not tag correctly

  #114 opened a year ago by Binbose
  7

• division by zero error for scalar params

  #142 opened a year ago by szbernat
  1

• Quick question on "layer_tag_vjp"

  #95 opened 2 years ago by connection-on-fiber-bundles
  2

• How to use kfac to train two probabilistic models jointly?

  #53 opened 2 years ago by wangleiphy
  3

• TypeError: 'ShapedArray' object is not iterable

  #10 opened 2 years ago by ltz0120
  4

• Unpack Error when using KFAC with block-diagonal for Dense networks

  #15 opened 2 years ago by joeryjoery
  4