haha "full rewrite in rust" is such a nerd thing! But I get the motivation.
The speeds mentioned, both from datago but also torch, are pretty abysmal though! We usually got 1-2k img/sec. However, as long as it creates a batch faster than one train step, who cares.