Add adaptive-step ODE solver and VJP calculation with JAX control flow operators for fast compilation
Add adaptive-step ODE solver and VJP calculation with JAX control flow operators for fast compilation

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