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

[nnx] add more Variable proxy methods #4170

Merged
merged 1 commit into from
Sep 5, 2024
Merged

Conversation

cgarciae
Copy link
Collaborator

@cgarciae cgarciae commented Sep 4, 2024

What does this PR do?

Improves the support of Variable as a proxy over its .value property, attribute access, item access, function calling are now supported. Example:

w = nnx.Param(jnp.ones((2, 3)))
w.T.shape == (3, 2)

@cgarciae cgarciae changed the title [nnx] add Variable proxy methods [nnx] add more Variable proxy methods Sep 4, 2024
@copybara-service copybara-service bot merged commit aded9ac into main Sep 5, 2024
18 checks passed
@copybara-service copybara-service bot deleted the nnx-variable-proxy branch September 5, 2024 01:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants