Jit-able runtime assertions for JAX in NumPy style.
Primary LanguagePythonApache License 2.0Apache-2.0