GitHub repository: tillahoffmann/jax-ifnt
Primary language: Python
Description: Execute runtime assertions, indexing checks, and more if `jax` code is not traced.