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.

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

Avoiding Type Piracy in Lux #246

Merged
merged 5 commits into from
Mar 19, 2024
Merged

Avoiding Type Piracy in Lux #246

merged 5 commits into from
Mar 19, 2024

Conversation

avik-pal
Copy link
Contributor

@avik-pal avik-pal commented Feb 25, 2024

Once upon a time before package extensions, I had put in hacks in Lux to make several of the DL packages play well with ComponentArrays. We moved some of them here before, but some still remained and it seems bad to have a different behavior for these packages before and after Lux is loaded.

I am moving those patches here. Let, me know if there is an objection regarding any of them.

TODOs

  • ReverseDiff Tracked arrays more functions
  • TruncatedStacktraces
  • rrule for ComponentArray(::NamedTuple)
  • Functors
  • Optimisers
  • Tests

Fixes LuxDL/Lux.jl#482

@codecov-commenter
Copy link

codecov-commenter commented Feb 25, 2024

Codecov Report

Attention: Patch coverage is 73.52941% with 9 lines in your changes are missing coverage. Please review.

Project coverage is 74.09%. Comparing base (fac804d) to head (a4e5293).

Files Patch % Lines
ext/ComponentArraysReverseDiffExt.jl 50.00% 6 Missing ⚠️
src/compat/chainrulescore.jl 72.72% 3 Missing ⚠️

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files
@@            Coverage Diff             @@
##             main     #246      +/-   ##
==========================================
+ Coverage   73.88%   74.09%   +0.20%     
==========================================
  Files          23       25       +2     
  Lines         743      772      +29     
==========================================
+ Hits          549      572      +23     
- Misses        194      200       +6     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@avik-pal
Copy link
Contributor Author

@jonniedie let me know if there is any patch you don't want here.

@avik-pal
Copy link
Contributor Author

Bump

Copy link
Owner

@jonniedie jonniedie left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for doing this! Sorry it took me so long to get to.

@jonniedie jonniedie merged commit 0fe0f85 into jonniedie:main Mar 19, 2024
7 of 8 checks passed
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.

ComponentArrays type piracy
3 participants