We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
For instance, add
from jax.config import config config.update("jax_enable_x64", True)
But, when using F64, XLA is not available.
See here for more detail.