Update: It went up Wednesday morning instead! Adding this turned up a bug in bayeux’s structural vi from TFP implementation, but it should now work generally.
Next on the list is trying to add some VI from numpyro.
Probably not (yet) a good idea, but you could delete lots of pymc/sampling/jax.py and replace it with bayeux now 
(just don’t delete the jaxify functions that bayeux uses…)