[mrdoops]

Yep and this approach also allows languages like Julia and Elixir to compile their expressions into valid compute graphs that target JAX/XLA. That polyglot capability opens up cutting edge machine learning into quite a bit more ecosystems with another level of capabilities in distribution and fault tolerance as is the case with Elixir + Nx.

[servytor]

Could you or someone elaborate more on how other languages can hook into JAX/XLA?

[nestorD]

Julia has XLA.jl [0] which interoperate with their deep-learning stack and Elixir has NX [1] which is higher level (basically JAX but in Elixir). I would love to see someone do something like that in Rust...

[0]: https://github.com/FluxML/XLA.jl

[1]: https://github.com/elixir-nx/nx/tree/main/nx