Just wondering if this is possible at the moment, it seems the old brax.io.image module is now gone in brax v2.

Moreover, if there is rendering support, are they GPU parallelizable or is it just state now (with single thread rendering for an eval env)

Thanks!