Rust impl

[thewh1teagle]

I would like to use it from Rust,
I saw that google jax has only python library
So I'm wondering how hard will be write it in Rust?
Or if there is similar algorithm already implemented in rust we can use?
Final goal is use it with vibe