PR #4374 Add `flax.nnx.eval_shape` docstring

Add `flax.nnx.eval_shape` docstring #4374

copybara-service merged 2 commits into google:main from 8bitmp3:add-nnx-eval_shap-docstring

Add flax.nnx.eval_shape docstring

6b5b3006

8bitmp3 requested a review from

levskaya 1 year ago

8bitmp3 requested a review from

cgarciae 1 year ago

8bitmp3 requested a review from

IvyZX 1 year ago

8bitmp3 assigned

8bitmp3 1 year ago

Add flax.nnx.eval_shape docstring

ddd7847b

IvyZX approved these changes on 2024-11-12

cgarciae added pull ready

copybara-service merged f265a5e2 into main 1 year ago

8bitmp3 deleted the add-nnx-eval_shap-docstring branch 1 year ago

Reviewers

IvyZX

levskaya

cgarciae

Assignees

8bitmp3

Labels

pull ready

Milestone

No milestone