flax
feat(nnx): add object-oriented pooling layers and GlobalAveragePool
#5201
Open
Open
feat(nnx): add object-oriented pooling layers and GlobalAveragePool #5201
divye-joshi
wants to merge
3 commits into
google:main
from
divye-joshi:add-pooling-modules
feat(nnx): add pooling layers (MaxPool, AvgPool, GlobalAveragePool)
9e1bc7e0
samanklesaria
commented
on 2026-02-02
refactor(nnx): remove GlobalAveragePool per review
b58b18ba
samanklesaria
approved these changes
on 2026-02-03
ci: trigger rebuild
874818d9
Login to write a write a comment.
Login via GitHub