flax

b3440228 - Merge pull request #2451 from dslisleedh:avg_pool_countpad