There are torch bindings that are used by some users but what I personaly would like is a JAX clone built on top of Rust. I see a way to do it [0] but... I alr

There are torch bindings that are used by some users but what I personaly would like is a JAX clone built on top of Rust. I see a way to do it [0] but... I already have a PhD to finish. [0]: a macro compiling functions into an intermediate representations that are transformed with const functions (gradient computation) at compile time and jitted into XLA at runtime.