jax
4413bb8a - [jax2tf] Do not use jax.random.PRNGKey before in primitive harness (#4211)

Commit

5 years ago

[jax2tf] Do not use jax.random.PRNGKey before in primitive harness (#4211) We cannot execute JAX functions before the program is initialized

References

Author

gnecula

Parents