flax
Added preferred_element_type arg to nnx.Linear, nnx.Conv, nnx.Einsum
#4920

Merged

Added preferred_element_type arg to nnx.Linear, nnx.Conv, nnx.Einsum #4920

copybara-service merged 1 commit into main from add-preferred_element_type-arg

vfdev-5 commented on 2025-08-29

Added preferred_element_type arg to nnx.Linear*, nnx.Conv*, nnx.Einsum

ab2efb97

vfdev-5 force pushed from c44e2302 to ab2efb97 160 days ago

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

vfdev-5 requested a review from

cgarciae 160 days ago

vfdev-5 commented on 2025-08-29

cgarciae approved these changes on 2025-09-03

cgarciae added pull ready

copybara-service merged 5e82ac8d into main 154 days ago

copybara-service deleted the add-preferred_element_type-arg branch 154 days ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone