Hadi Dolatabadi

Hi, Related to this issue, if you try a large network (e.g. the Glow architecture for CIFAR-10), then you may encounter an error in the middle of training which says:...

Hi. Encountered the same error today. Apparently, this is due to JAX not supporting boolean indexing, e.g., see [here](https://github.com/data-apis/array-api/issues/84). I was curious to see if you have found a solution...

Hi All, I have been using the `Python Optimal Transport` library. I want to define a loss function that iterates over every sample in my batch and calculates the `sinkhorn`...

Thanks @rflamary! I wanted to join the POT Slack, but unfortunately it seems that the workspace invite link hasn't been shared. Could you send me the POT Slack invite? Thanks.

Hi! This is apparently because of the version mismatch in FrEIA. You can take a look at [this question](https://github.com/hmdolatabadi/AdvFlow/issues/5) and use my answer to deal with this issue.

Thanks! I am running my experiments with a slightly modified version of ResNet-32, where the `BatchNorm2d` is done before the shortcut, and I am not sure whether this is the...