Search results

Unexpected change in parameter registration from 'Orphan' to 'Generic' causes training failure in kfac_jax 0.0.7

Status: Open.

#344 In google-deepmind/kfac-jax;

· DanChai22 opened on Jun 3, 2025

Sobolev training in KFAC

Status: Open.

#342 In google-deepmind/kfac-jax;

· rbvh opened on May 31, 2025

One problem about kfac_jax

Status: Open.

#338 In google-deepmind/kfac-jax;

· Yongda1 opened on May 20, 2025

AttributeError raised by tag_graph_matcher.py when using Ferminet

Status: Open.

#314 In google-deepmind/kfac-jax;

· danielepremarini opened on Feb 9, 2025

TypeError: unhashable type: 'Literal'

Status: Open.

#310 In google-deepmind/kfac-jax;

· DanChai22 opened on Jan 13, 2025

use of core.unsafe_get_axis_names_DO_NOT_USE which no longer exists

Status: Open.

#285 In google-deepmind/kfac-jax;

· svandenhaute opened on Oct 31, 2024

Bug: Jax 0.4.13 support

Status: Open.

#279 In google-deepmind/kfac-jax;

· arnon-1 opened on Oct 23, 2024

Incorrectly pytree recognition by KFAC optimizer

Status: Open.

#273 In google-deepmind/kfac-jax;

· Uernd opened on Oct 4, 2024

Model parameters being marked as orphan when using KFAC optimizer

Status: Open.

#270 In google-deepmind/kfac-jax;

· Uernd opened on Sep 22, 2024

Can this be used for Laplace approximation?

Status: Open.

#224 In google-deepmind/kfac-jax;

· blackblitz opened on May 9, 2024

Add Support for KFAC Optimization in LSTM and GRU Layers

Status: Open.

#188 In google-deepmind/kfac-jax;

· neuronphysics opened on Nov 22, 2023

Using K-FAC with physics-based losses

Status: Open.

#186 In google-deepmind/kfac-jax;

· SNMS95 opened on Nov 14, 2023