Skip to content

Add a note about jax.pmap when leading dim is smaller than num devices.#2949

Merged
jekbradbury merged 1 commit intojax-ml:masterfrom tomhennigan:changelist/309724606May 4, 2020
+10

Commits