Conversion of algorithms into Rust

Hi deepmind team,

This is truly amazing work! Great job and thank you so much for publishing these findings. As a developer I'm keen to make use of these algorithmic improvements. I've started to port the algorithms to Rust here: https://github.com/drbh/simd-alphatensor-rs by programmatically generating the code - however I'm not sure how to best implement/share these improvements. As the creators of such work do you have any suggestions for packaging these into a library? Are there any implementation considerations that should be taken into account when building a library around this work?

Thanks again!!