How to fix the RuntimeError: builtin cannot be used as a value (with a dict) when creating checkpoints from jit in pytorch?

Question

I want to compile arbitrary nn modules using jit.

In my code I compare my value to a dict and it throws an error.

        """
        if typ