Source code for eztaox

try:
    from importlib.metadata import PackageNotFoundError, version  # type: ignore
except ImportError:  # pragma: no cover
    from importlib_metadata import PackageNotFoundError, version  # type: ignore
try:

[docs]
    __version__ = version(__name__)

except PackageNotFoundError:  # pragma: no cover
    __version__ = "unknown"

import jax

jax.config.update("jax_enable_x64", True)