Showing content from https://github.com/jax-ml/jax/issues/6129 below:
no constant handler for IntEnum · Issue #6129 · jax-ml/jax · GitHub
Skip to content Navigation Menu
Saved searches Use saved searches to filter your results more quickly
Sign up You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session. Dismiss alert Additional navigation options
no constant handler for IntEnum #6129
Description
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.
You can’t perform that action at this time.
RetroSearch is an open source project built by @garambo
| Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.3