Yih-Dar c15bb3fe19 [Fix doc example] fix missing import jnp (#15291) ...

* fix missing import jnp

* Fix missing jax and k=1

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

2022-01-24 14:54:23 +01:00

..

adding_a_new_example_script

Happy New Year! (#15094)

2022-01-10 12:05:57 -05:00

adding_a_new_model

[Fix doc example] fix missing import jnp (#15291)

2022-01-24 14:54:23 +01:00