flax
Added kernel_metadata/bias_metadata args to nnx layers
#5074

Merged

Added kernel_metadata/bias_metadata args to nnx layers #5074

copybara-service merged 2 commits into main from params_metadata_args

vfdev-5 commented on 2025-11-07

vfdev-5 force pushed from 1af973b6 to f385e257 87 days ago

vfdev-5 marked this pull request as ready for review 87 days ago

cgarciae commented on 2025-11-07

vfdev-5 force pushed from f385e257 to a5426a9e 86 days ago

vfdev-5 force pushed from a5426a9e to 67864643 86 days ago

vfdev-5 force pushed from 67864643 to 485424c6 86 days ago

cgarciae approved these changes on 2025-11-10

cgarciae added pull ready

Added kernel_metadata/bias_metadata args to nnx layers

c6036553

Implement *_metadata to other layers

f320a5c2

vfdev-5 force pushed from 485424c6 to f320a5c2 82 days ago

vfdev-5 requested a review from

cgarciae 82 days ago

cgarciae approved these changes on 2025-11-12

copybara-service merged b08cb206 into main 81 days ago

copybara-service deleted the params_metadata_args branch 81 days ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone