Skip to content

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