Skip to content

Conversation

@avik-pal
Copy link
Member

@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
Member Author

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

@avik-pal
Copy link
Member Author

Bump

Copy link
Collaborator

@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 SciML:main Mar 19, 2024
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