o qo™i8ã@sdZddlmZdS)z„Fiddle extensions to handle JAX code more elegantly. Currently this just affects codegen, graphviz, and other debugging functions. é)ÚenableN)Ú__doc__Úfiddle._src.extensions.jaxr©rrúI/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/extensions/jax.pyÚs