comment on joining staged jaxprs in partial evaluation of conditionals
comment on joining staged jaxprs in partial evaluation of conditionals

google/jax,tensorflow/probability,google/jax,google/jax,google/jax,tensorflow/probability