Nice. I helped write the original Python implementation, so thanks for building this. STARK proof generation is definitely a good candidate for GPU acceleration.