Peter Luo

Results 1 issues of Peter Luo

The lastest Stax in JAX should be jax.example_libraries.stax as documented here: https://jax.readthedocs.io/en/latest/jax.example_libraries.stax.html