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

Flax dataclasses do not support new-style PRNGKeyArray in serialisation #4130

Open
PhilipVinc opened this issue Aug 19, 2024 · 0 comments
Open

Flax dataclasses do not support new-style PRNGKeyArray in serialisation #4130

PhilipVinc opened this issue Aug 19, 2024 · 0 comments
Labels
Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)

Comments

@PhilipVinc
Copy link
Contributor

New-style PRNGKeyArray cannot be serialised by flax.
This is inconveniente as one has to manually override the serailization/deserialization methods automatically generated by flax's data class to wrap/unwrap those fields..

Would it be possible to support natively PRNGKeyArray (which are technically leafs) within flax.serialization?
@cgarciae cgarciae added the Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required) label Aug 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
Priority: P1 - soon Response within 5 business days. Resolution within 30 days. (Assignee required)
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@PhilipVinc @cgarciae