2 ms·
Could you or someone elaborate more on how other languages can hook into JAX/XLA?
by servytor 5y ago
Could 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