jax
0f7b6444 - Enable XLA SPMD partitioning by default. (#3151)

Commit

5 years ago

Enable XLA SPMD partitioning by default. (#3151) This option is needed for sharded_jit. Future APIs may use MPMD partitioning instead.

References

Author

skye

Parents