jax
2f41a251 - Add num_partitions argument to sharded_jit (#3392)

Commit

5 years ago

Add num_partitions argument to sharded_jit (#3392)

References

#3392 - Add num_partitions argument to sharded_jit

Author

skye

skye

Parents

Loading