Replace jnp.row_stack with jnp.vstack

probml / dynamax

State Space Models library in JAX

https://probml.github.io/dynamax/

MIT License

634 stars 70 forks source link

Replace jnp.row_stack with jnp.vstack #340

Closed jakevdp closed 10 months ago

jakevdp commented 10 months ago

Why? Following NumPy's NEP 52 API cleanup, np.row_stack is being removed, as it is a simple alias of np.vstack. JAX is following suit by deprecating jnp.row_stack in https://github.com/google/jax/pull/17233. Note that jnp.vstack is an identical replacement.