Add a docstring for jax.set_mesh - SemanticDiff

Commit

46 days ago

Add a docstring for jax.set_mesh PiperOrigin-RevId: 829112978

References

#33156 - Add a docstring for jax.set_mesh

#33157 - Resolve undefined behavior in bitshift unit test

#33186 - Make nvidia version data optional for ROCm builds

Author

yashk2810

yashk2810

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading