Raphaël Boige

Hello, and thanks for the answer and the clarification. There is still something I don't get. As far as I understand the `MVC` argument is set to `True` is the...

> Please provide benchmarking code for cfrx side if example is not full picture (for example doesn't use vmap extensively idk). > I confirm the code used for benchmarking is...

I see, my minimal reproduction was ambiguous. Sorry for that. I figured out that it depends on the order of the decorator @jaxtyped. This fails: ```python3 import beartype import flax...