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