fix(inject_hyperparams): treat plain Python ints as static to fix adafactor jit (#412)#1686

nileshpatil6 wants to merge 1 commit into

google-deepmind:maingoogle-deepmind/optax:mainfrom

nileshpatil6:fix/adafactor-inject-hyperparams-jitnileshpatil6/optax:fix/adafactor-inject-hyperparams-jitCopy head branch name to clipboard