jax
0b46a236 - Update Pallas distributed tutorials with jax.make_mesh

Commit

1 year ago

Update Pallas distributed tutorials with jax.make_mesh

References

#24434 - Update Pallas distributed tutorials with jax.make_mesh

Author

justinjfu

justinjfu

Committer

justinjfu

justinjfu

Parents

Loading