from lapjax.experimental.jax2tf.examples import keras_reuse_main as keras_reuse_main
from lapjax.experimental.jax2tf.examples import saved_model_main_test as saved_model_main_test
from lapjax.experimental.jax2tf.examples import mnist_lib as mnist_lib
from lapjax.experimental.jax2tf.examples import keras_reuse_main_test as keras_reuse_main_test
from lapjax.experimental.jax2tf.examples import saved_model_main as saved_model_main
from lapjax.experimental.jax2tf.examples import saved_model_lib as saved_model_lib
import sys, importlib
from lapjax.lapsrc.wrapper import _wrap_module
_wrap_module(importlib.import_module(__name__.replace('lapjax', 'jax')), 
             sys.modules[__name__])
