flax
feat(nnx): add preferred_element_type to attention.py and recurrent.py, out_sharding to recurrent.py
#5179
Open
Open
feat(nnx): add preferred_element_type to attention.py and recurrent.py, out_sharding to recurrent.py #5179
aarushisingh04
wants to merge
2 commits into
google:main
from
aarushisingh04:nnx-technical-support
samanklesaria
commented
on 2026-01-08
samanklesaria
commented
on 2026-01-08
samanklesaria
commented
on 2026-01-08
aarushisingh04
changed the title
samanklesaria
approved these changes
on 2026-01-13
chapman20j
commented
on 2026-01-22
feat(nnx): add preferred_element_type to attention.py and recurrent.p…
be9ea482
Simplify docstring for preferred_element_type
0d936835
cgarciae
commented
on 2026-02-02
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub