Rewrite in Rust^W Zig^W Jai

[valignatev]