Skip to content

Instantly share code, notes, and snippets.

@mwitiderrick

mwitiderrick/nan.py

Created July 11, 2022 12:04
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/788f54c0d0b865dc98f3f02491929da7 to your computer and use it in GitHub Desktop.
Save mwitiderrick/788f54c0d0b865dc98f3f02491929da7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nan.py
from jax.config import config
config.update("jax_debug_nans", True)
jnp.divide(0.0,0.0)
# FloatingPointError: invalid value (nan) encountered in div
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment