Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add JAX backend support #13

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Add JAX backend support #13

wants to merge 2 commits into from

Conversation

bonneted
Copy link

No description provided.

@lululxvi lululxvi requested a review from mitchelldaneker June 17, 2024 15:28
@bonneted
Copy link
Author

I've got the same final parameters as the ones in the paper :

Epoch E tp ti td k Rm a1 C1 C2 C4 C5 Ub U0 Um Rg alpha beta
1010000 0.20086897812616378 5.984280323717105 101.07973763650493 11.9777607693349 0.008332535739606049 208.50174547489897 6.593664003041278 301.0031941572703 34.69381362708656 79.07846369570885 25.952992080032526 71.5795274689672 3.99920640001627 89.31458070648316 179.61120094478127 7.538939453290313 1.778194872565763

@lululxvi
Copy link
Member

Is this ready to merge?

@bonneted
Copy link
Author

The code relies on this DeepXDE PR that isn't merged yet: lululxvi/deepxde#1775
I've just updated the code of DeepXDE PR, please let me know if you'd like any more changes.

I've tested the code again and it's working as intended.
So whenever the other PR is accepted, this one should be ready to merge.

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

@Dchharlow4171717 Dchharlow4171717 Dchharlow4171717 approved these changes

@mitchelldaneker mitchelldaneker Awaiting requested review from mitchelldaneker

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@bonneted @lululxvi @Dchharlow4171717