Julia's Broadcast vs Jax's vmap

Coming back to this down the line, with XLA.jl archived, is there any packages of note in this direction (beyond the growth of Zygote.jl)? To me, this would be a great Julia package (coming back into the Julia ecosystem having spent time in ML w/ python).

One promising direction being worked on along those lines would be Reactant.jl. It’s still immature, but it is under active development.

3 Likes

Yes Reactant is the current development going on, along with some other efforts for MLIR that come and go. But I would bet on Reactant being the one to ship first in a usable state.

4 Likes