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

Housekeeping + Use Faster Nested AD #152

Merged
merged 6 commits into from
Apr 25, 2024
Merged

Housekeeping + Use Faster Nested AD #152

merged 6 commits into from
Apr 25, 2024

Conversation

avik-pal
Copy link
Member

@avik-pal avik-pal commented Apr 24, 2024 •
edited
Loading

Checklist

@avik-pal
Copy link
Member Author

Documentation build is fine locally, once LuxDL/Lux.jl#598 is merged.

@avik-pal
Copy link
Member Author

Okay need to address LuxDL/LuxLib.jl#34 ConvTranspose for this to work

@avik-pal avik-pal force-pushed the ap/unbreak_nested_ad branch 2 times, most recently from a4d8d2d to cd9f5f4 Compare April 24, 2024 22:54
@avik-pal
Copy link
Member Author

avik-pal commented Apr 24, 2024 •
edited
Loading

It works 🎉. Nested AD works with Lux and is actually efficient now 🎉

@avik-pal avik-pal merged commit 47bcaa9 into main Apr 25, 2024
5 of 8 checks passed
@avik-pal avik-pal deleted the ap/unbreak_nested_ad branch April 25, 2024 18:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

Unbreak Nested AD
1 participant
@avik-pal