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

bridge module with linen submodule #4604

Merged

copybara-service merged 1 commit into google:main from IvyZX:linx-nn

Mar 7, 2025

+130 −29

Collaborator

IvyZX commented Mar 6, 2025

What does this PR do?

Allow Linen modules to become submodules of bridge.Module, supporting the mutable arg and allow custom method call.
Cleaned up ToNNX():
- removed unneeded linen_attributes
- prevent RNG reuse during init
- auto-recognize mutable arg if top level bridge.Module context exists.

IvyZX force-pushed the linx-nn branch from dd45d06 to 523c5c5 Compare

March 6, 2025 01:24

IvyZX requested a review from cgarciae

March 6, 2025 04:03


          bridge module with linen module

300494b

IvyZX force-pushed the linx-nn branch from 523c5c5 to 300494b Compare

March 6, 2025 04:04

cgarciae approved these changes

View reviewed changes

cgarciae added the pull ready label

copybara-service bot merged commit d59132d into google:main

16 checks passed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels