zombie-einstein

icon indicating a website link https://zombie-einstein.github.io/

icon location Research engineer interested in machine-learning and agent-based modelling Take a look at my blog https://zombie-einstein.github.io/

Results 4 comments of zombie-einstein

Error when the for i loop don't start with 0

Think this is the same issue as this https://github.com/jeremiecoullon/jax-tqdm/issues/12 , comments there may help you

XlaRuntimeError: Outside call <jax.experimental.host_callback._CallbackWrapper object ... >

> I think the following is related. On jax 0.4.28 I get `KeyError: 0` in `_update_tqdm` from `tqdm_bars[0].update(arg)`, even when iterations begin at 0. I've actually been meaning to file...

XlaRuntimeError: Outside call <jax.experimental.host_callback._CallbackWrapper object ... >

> > > I think the following is related. On jax 0.4.28 I get `KeyError: 0` in `_update_tqdm` from `tqdm_bars[0].update(arg)`, even when iterations begin at 0. > > > >...

XlaRuntimeError: Outside call <jax.experimental.host_callback._CallbackWrapper object ... >

I wonder if the `ordered` flag [here](https://jax.readthedocs.io/en/latest/_autosummary/jax.debug.callback.html#jax-debug-callback)is what we are looking for?