You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
In the long term, I believe we need to have a jaxpr level graph rewriting tool, we can then introduce algebraic relationships that works more generally for any nested nabla operations.
The text was updated successfully, but these errors were encountered:
mavenlin
changed the title
Add laplace operator
Add laplacian operator
Dec 7, 2023
According to https://github.com/y1xiaoc/fwdlap/, jet is not the most efficient option for laplace calculation, a more efficient way would be to reduce at every op.
Immediate solution
Laplace operator is important for computing the kinetic energy, currently we have to use two nabla operators together with the trace. For example
in the transpose rule of nabla,
https://github.com/sail-sg/autofd/blob/main/autofd/operators/operators.py#L773
in mgga functional of
jax_xc
,https://github.com/sail-sg/jax_xc/blob/main/gen_repo/experimental.jinja#L66
While we can do better with the jet rules.
In the long term
In the long term, I believe we need to have a jaxpr level graph rewriting tool, we can then introduce algebraic relationships that works more generally for any nested nabla operations.
The text was updated successfully, but these errors were encountered: