How to Implement MaxPooling2D with Ceil Mode in Flax?

[π12Qα]

What is the equivalent code for maxpooling2d with ceil mode in PyTorch?

So for example in PyTorch we can code like this :
• nn.MaxPool2d(3, stride=2, padding=0, ceil_mode=True)
But in Flax, the framework only supports four parameters such as input, window_size, strides, and padding.