Skip to content

Instantly share code, notes, and snippets.

@mwitiderrick

mwitiderrick/float.py

Created July 11, 2022 12:13
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mwitiderrick/f21aee9617b44fc21f4a94c7a0df709a to your computer and use it in GitHub Desktop.
Save mwitiderrick/f21aee9617b44fc21f4a94c7a0df709a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
float.py
x = jnp.float64(1.25844)
# /usr/local/lib/python3.7/dist-packages/jax/_src/numpy/lax_numpy.py:1806: UserWarning: Explicitly requested dtype float64 requested in array is not available, and will be truncated to dtype float32. To enable more dtypes, set the jax_enable_x64 configuration option or the JAX_ENABLE_X64 shell environment variable. See https://github.com/google/jax#current-gotchas for more.
# lax_internal._check_user_dtype_supported(dtype, "array")
# DeviceArray(1.25844, dtype=float32)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment