Skip to content

Conversation

@facusapienza21
Copy link
Member

No description provided.

@facusapienza21 facusapienza21 marked this pull request as ready for review November 4, 2025 23:38
@facusapienza21
Copy link
Member Author

I open this PR to check whether MoonCake can be use to compute the gradients more efficiently in SphereUDE. With this core, the benchmark for the following adjoint methods give the following results:

sensealg_types = [
    GaussAdjoint(autojacvec = ReverseDiffVJP(true)),
    InterpolatingAdjoint(autojacvec = ReverseDiffVJP(true)),
    QuadratureAdjoint(autojacvec = ReverseDiffVJP(true)),
    GaussAdjoint(autojacvec = MooncakeVJP()),
    InterpolatingAdjoint(autojacvec = MooncakeVJP()),
    QuadratureAdjoint(autojacvec = MooncakeVJP()),
    SphereBackSolveAdjoint()
]
image

I don't observe any incredible increase in efficiency due to Mooncake, but also I am not sure I am using right. All this benchmarks are done without any regularization, so nested AD is not really required.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants