It seems like it would be easy enough to train the model to improve performance too once you have it writing correct code. Write N problems and N performance te

It seems like it would be easy enough to train the model to improve performance too once you have it writing correct code. Write N problems and N performance test suites - then train by self-play on those problems until you're writing high performance code.