Introduction to pjit

Introduction to pjit#

This content is no longer relevant, because pjit() and jit() have been merged into a single unified interface. For an updated guide to compiling and executing JAX functions in multi-host or multi-core environments, see Distributed arrays and automatic parallelization.