no constant handler for IntEnum

[mattjj]

import enum
import jax

class Foo(enum.IntEnum):
  bar = 1
  baz = 2


@jax.pmap
def f(_):
  return Foo.bar

f(jax.numpy.arange(1))

The same works with jax.jit instead of jax.pmap.