f-dangel / curvlinops

92%

main: 94%

LAST BUILD ON BRANCH fisher-mc

branch: fisher-mc

CHANGE BRANCH

x

Reset

Sync Branches

• fisher-mc
• activation-hessian
• activation-hessian-linop
• adjoint
• adjoint-jacobians
• adjoint-submatrix
• alt-ekfac
• always-full-tests
• arbitrary-inputs
• bcewithlogitsloss
• bug-dtype-check-deterministic
• bug-kfac-device-change
• ci-update
• composition-rules
• curvature-linop
• deprecate-pkg-resources
• deprecate-python38
• development
• diagonal
• doc-eigenvalues
• ef-linop
• ekfac
• empirical-fisher
• eva
• example-benchmark
• example-fisher-mc
• example-trace
• expand-reduce
• faster-ef
• faster-fisher-mc
• fisher-averaging
• fisher-linop
• fix-coveralls-badge
• fix-device
• fix-device-errors
• fix-enum
• fix-inverse-kfac
• fix-kfac-scale
• fix-markdown-syntax
• fix-mc-tests
• fix-mcfisher-scale
• fix-python-badge
• fix-scale
• fix-typo
• flake8-to-ruff
• foof
• frobenius-norm
• functional-estimators
• functional-ggnmp
• ggn-linop
• hessian-blocks
• hessian-linop
• high-dim-outputs
• hutchinson
• hutchpp
• improve-benchmark-results
• improve-frobenius
• improve-spectral-density
• improve-tests
• improve-visual-tour
• issue-138-improve-links
• jacobian
• jacobian-linop
• jacobian-transposed
• joint-weight-and-bias
• kfac
• kfac-arbitrary-params-order
• kfac-ce-loss
• kfac-conv2d
• kfac-damping
• kfac-ef
• kfac-ema
• kfac-inv
• kfac-linop
• kfac-state-dict
• kfac-torch-only
• kfac-trace
• kfac-type-2
• kfac-with-inplace-activations
• lanczos-cache
• load-state-dict-model-func
• logo
• lsmr-inverse
• main
• matmat-list-shapes
• matvec-to-matmat
• neumann-inverse
• neumann-inverse-example
• no-cpu
• only-bias-or-weight
• papyan20traces-density
• parallelize-grad-output
• parallelize-sample-grad-output
• parallelize-tests
• partial-boundary-specification
• power-iteration-comparison
• pydocstyle-coverage
• pytorch-inverse
• pytorch-linop
• reduce-scipy
• ref-internal-mapping
• remove-kfac-side-effect
• remove-loss-average
• remove-to_device
• remover-iterator-conversion
• retry-kfac-inv
• runame-patch-1
• source-button
• specify-num-data
• submatrices
• submatrix-example
• submatrix-pytorch
• test-consecutive-matmats
• test-regression
• update-black
• update-changelog
• update-ci
• update-github-actions
• use-built-in-functorch
• use-pyproject-toml
• visual-tour
• weighted-ce-kfac
• xdiag
• xtrace

[FIX] Docstring, extract number of classes