Rust version of jax

[any35]

Hello, I want to know if there are plans to implement rust version of jax? thans.

[mattjj]

Thanks for the question! Actually, this came up in #5938, so I'll repeat an interesting factoid: @froystig used to work on Rust, more than a decade ago! And while doing that at Mozilla he met another JAX cofounder @cdleary, who I think was working on JavaScript JITs at the time.

Having a Rust version sounds like a cool idea, but it's out of scope for us right now.

[jgsimard]

@mattjj Hi, is it in scope now or in the future?

[mattjj]

Not in scope in the foreseeable future, sorry.

[Oceania2018]

What about expose the C interface for other languages to bind?

[rjzak]

I too would be interested in Rust or C for Jax, so that Jax models could be used outside of Python.

[cksac]

Hi, I am currently working on implement a ML framework similar to JAX in Rust, https://github.com/cksac/rai.
Still in early stage, it now use Candle as backend, I am planning to add XLA/IREE backend later

[rjzak]

Starred, thanks!