flax
27b233d7 - Add support for transforming methods that are not user functions (e.g.: partial functions).

Commit

4 years ago

Add support for transforming methods that are not user functions (e.g.: partial functions). This also fixes a bug where a partial object is always assumed to wrap a Module constructor

References

#1623 - Add support for transforming methods that are not user functions (e.g…

Author

jheek

Committer

jheek

Parents

8842c69f

flax 27b233d7 - Add support for transforming methods that are not user functions (e.g.: partial functions).

flax
27b233d7 - Add support for transforming methods that are not user functions (e.g.: partial functions).