1
91%
main: 92%

Ran 09 Mar 2026 02:54AM UTC

Files 47

Run time 1s

Badge

Embed ▾

Committed 09 Mar 2026 02:16AM UTC coverage: 91.259% (-0.02%) from 91.274%

Job # 22835615977.1

Build Type

push

github

Committed by f-dangel

Commit Message

[REF] Split vmap data args into X and loss_args, simplify dims to -1

Separate model input X from loss function arguments loss_args in
make_ggn_vector_product and all downstream callers. This makes the
data flow explicit: f consumes X, c consumes loss_args. Adding extra
loss args (e.g. generator for MC) no longer changes vmap in_dims.

Also replace tuple(p.ndim for p in params) with scalar -1 in all vmap
in_dims/out_dims. A scalar -1 broadcasts to "last axis" per tensor
leaf in the pytree, which is the trailing matrix-column dimension.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Coverage Stats

2203 of 2414 relevant lines covered (91.26%)

0.91 hits per line

f-dangel / curvlinops / 22835615977 / 1
91%
main: 92%

README BADGES
x

Markdown

Textile

RDoc

HTML

Rst

Source Files on job 22835615977.1

f-dangel / curvlinops / 22835615977 / 1 91% main: 92%

README BADGES x

Markdown

Textile

RDoc

HTML

Rst

Source Files on job 22835615977.1

f-dangel / curvlinops / 22835615977 / 1
91%
main: 92%

README BADGES
x