Skip to main content
Ctrl+K

Parallel computation#

Note

This is a placeholder for a section in the new JAX tutorials.

For the time being, you may find some related content in the old documentation:

  • Using JAX in multi-host and multi-process environments

  • Distributed arrays and automatic parallelization