Owen L

When you say you want to keep the state of the helper qubits do you mean the parameters or the exact wavefunction at the end of the execution? Maybe if...

Mid circuit measurements (and qubit resets) are hard to do sometimes because they are can be non-unitary. I might do something like just keep track of your operations on the...

Well you can do that in cirq by simply using the reset op. In TFQ that is not a syntheziable op (you would get the following error if you tried...

Any updates on this or should it be closed?

> Experienced jax users could point that Equinox stateful operation (e.g. for kv cache, batchnorm, ...) is based around a jax bug. This would break Equinox and downstream libraries (e.g....

> * Avoid listing/recommending downstream projects > * List a small set of downstream projects, based on some non-objective criterion such as "what Jake likes" > * List a small...

To be extra clear, if you make `count_initial = 0` you see the same speed for sharding and for pmap. Only when this varies per shard does this slow down

Also, if you replace the body and do ```python new_t, new_y, theta = inner_loop_body((t, y, theta)) #new_t, new_y, _ = inner_while_loop(inner_loop_cond, inner_loop_body, (t, y, theta)) return (new_t, new_y, theta, count...

@patrick-kidger you mentioned in https://github.com/patrick-kidger/diffrax/issues/407 that you suspect this is within XLA, do you have any advice on how to approach that? I haven't investigated an XLA system this complex...

I have limited NLP experience, but I seem to have 2 possible understandings (both of which are conditioned on some input) 1) You have a collection of circuits that have...