jax.CompilerEffortLevel

jax.CompilerEffortLevel#

class jax.CompilerEffortLevel(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)[source]#

Effort level enumeration for XLA.

Used to specify the degree to which the XLA compiler should optimize for runtime performance or memory fitting, as described in https://openxla.org/xla/effort_levels. Higher effort levels will expend more compile time but should yield better results in the respective dimension.

__init__(*args, **kwds)#

Attributes

UNKNOWN

O0

O1

O2

O3