jax
83ca7c33 - Don't create a mesh inside jit in sharded-computation tutorial

Commit

1 year ago

Don't create a mesh inside jit in sharded-computation tutorial PiperOrigin-RevId: 707288222

References

Author

yashk2810

Committer

Google-ML-Automation

Parents