Skip to content

Instantly share code, notes, and snippets.

@mwitiderrick

mwitiderrick/float64.py

Created July 11, 2022 12:18
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/684ee102f335ee83becba3e344e9c6c0 to your computer and use it in GitHub Desktop.
Save mwitiderrick/684ee102f335ee83becba3e344e9c6c0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
float64.py
# set this config at the begining of the program
from jax.config import config
config.update("jax_enable_x64", True)
x = jnp.float64(1.25844)
x
# DeviceArray(1.25844, dtype=float64)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment