Parallelisation in neuraLQX

This section documents distributed parallelisation in neuraLQX. Use the JAX distributed pages below for current guidance.

JAX Distributed: Single Host

single_host.html

JAX Distributed: Multi-Host

multi_host.html

MPI (Deprecated)

mpi_deprecated/index.html

• JAX Distributed Parallelisation (Single Host)
• JAX Distributed Parallelisation (Multi-Host)
• MPI (Deprecated)