3 ms·
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 capa
by mrdoops 5y ago
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 5y agoCould you or someone elaborate more on how other languages can hook into JAX/XLA?
- nestorD 5y agoJulia 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 https://github.com/FluxML/XLA.jl [1]: https://github.com/elixir-nx/nx/tree/main/nx https://github.com/elixir-nx/nx/tree/main/nx