jax
d585aecc - print `pjit_p` as `"jit"` (in jaxpr, etc.)

Commit

1 year ago

print `pjit_p` as `"jit"` (in jaxpr, etc.) Less obscure this way. PiperOrigin-RevId: 777846049

References

Author

froystig

Committer

Google-ML-Automation

Parents