jorisSchaller

icon indicating a website link https://jsch.ch

icon location Zürich icon location MSc Artificial Intelligence, Experienced with building real and effective Machine Learning solutions. Currently intern @ai4sd (IBM research) Previously @EBU

Results 1 comments of jorisSchaller

NNX documentation missing pooling operations

Hey, I was also looking at the pooling, the `flax/core/nn/ __init__.py` file import the function from the old linen API. https://github.com/google/flax/blob/5d896bc1a2c68e2099d147cd2bc18ebb6a46a0bd/flax/core/nn/__init__.py#L38 But the entire file uses only https://github.com/google/flax/blob/5d896bc1a2c68e2099d147cd2bc18ebb6a46a0bd/flax/linen/pooling.py#L17-L19 We should...