Revision 4c423c36d0b4f0e05c29db507bd6bedd5e0f9e75 authored by Peter Hawkins on 18 January 2022, 18:41:43 UTC, committed by jax authors on 18 January 2022, 18:42:14 UTC
(check_jaxpr() is only used when debugging.)

Don't eagerly pretty print jaxprs: only do so if we are going to raise an error.
Don't eagerly form error messages. Delete typecheck_assert.

PiperOrigin-RevId: 422594126
1 parent e30b96c
Raw File
.gitignore
*.pyc
*.so
*.egg-info
*.whl
build/bazel*
dist/
.ipynb_checkpoints
/bazel-*
.jax_configure.bazelrc
/tensorflow
.DS_Store
.mypy_cache/
.pytype/
docs/build
docs/notebooks/.ipynb_checkpoints/
docs/_autosummary
.idea
.vscode
jax.iml

# virtualenv/venv directories
venv/
bin/
include/
lib/
share/
back to top