from jax import jit @jit def jitable_speciate(): pass