Have you looked at Can Nx replace Rust for number crunching? - Nx Forum - Elixir Programming Language Forum ?