Index

A

allgather() (in module mpi4jax) allreduce() (in module mpi4jax)

alltoall() (in module mpi4jax) auto_tokenize() (in module mpi4jax.experimental)

B

barrier() (in module mpi4jax)

bcast() (in module mpi4jax)

C

CUDA_ROOT

E

environment variable CUDA_ROOT

G

gather() (in module mpi4jax)

H

has_cuda_support() (in module mpi4jax)

R

recv() (in module mpi4jax)

reduce() (in module mpi4jax)

S

scan() (in module mpi4jax) scatter() (in module mpi4jax)

send() (in module mpi4jax) sendrecv() (in module mpi4jax)