trixi-framework / trixi-framework/Trixi.jl

Adding reverse mode AD support

Open
#827 9 comments 1 reaction 0 assignees View on GitHub
Dominant language
Julia
Stars
731
Forks
166
Avg merge
2d 18h
Merged PRs (30d)
25

Description

Larger PDE solves may benefit from a reverse mode AD and also helps with adding support for neural networks as surrogates or part of training the PDEs themselves, therefore I was wondering what it would take to add Zygote support here. It also comes with a forward mode which may be something that can be used as well.

Contributor guide

Open the contributing guide

Research direction

The issue proposes adding Zygote support for reverse-mode AD to Trixi.jl, with forward mode also mentioned for larger PDE solves and neural-network use cases. Start by locating the PDE solve entry points and assessing how Zygote would interact with them; done should mean a defined, working AD path for the stated use cases.

Written by the indexing model from the issue text.

Assessment

Tech stack
julia
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.