Chase Riley Roberts

I agree. I think just a straight port with `cdefs` should be the first step.

I've actually not used cython myself yet, so I'm not sure I know any gotchas ahead of time.

I like this idea. Would we basically map the `jax.lax.p*` primitives to `mpi4jax` primitives?

I think it sadly has come time to deprecate this library...

At this time no. Many of the things we were testing against are ready features of Keras, mainly the "make sure all variables get trained" has been in keras even...

@mattjj If you want to start looking at it go for it. All of the meat is there.

I need to add tests which requires backend support, but beyond that this is ready for full review.

I added a test.

Yes, I will add much more test coverage. Do we wish to also discuss the name `pcopy`? Now is the time to do so. Pinging @nouiz who had opinions to...

Good enough reasoning for me.