About environment version

[ppalantir]

Thanks for your awesome work!

Could you please provide the version of packages such as Python, tensorflow, jax, etc?

Thanks