mpi4jax was dead, then Claude resurrected it

I maintain a moderately successful open source project called mpi4jax that s useful in its small niche of people abusing JAX to do scientific computing. mpi4jax lets you run JAX code across multiple machines by calling MPI under the hood, without leaving the compiled execution graph. It s really a …